code: def test_features_extraction_spec_for_mode(self):
      dataset = make_golden_dataset()
  
      shared_feature_names = {
          "image/encoded",
          "variant/encoded",
          "alt_allele_indices/encoded",
          "variant_type",
          "sequencing_type",
      }
      self.assertEqual(
          shared_feature_names,
          set(
              dataset.features_extraction_spec_for_mode(
                  include_label_and_locus=False
              ).keys()
          ),
      )
      self.assertEqual(
          shared_feature_names.union({"label", "locus"}),
          set(
              dataset.features_extraction_spec_for_mode(
                  include_label_and_locus=True
              ).keys()
          ),
      )
  
imports: import math
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow import estimator
  from deepvariant import data_providers
  from deepvariant import dv_config
  from deepvariant import dv_constants
  from deepvariant import dv_utils
  from deepvariant import testdata
  from deepvariant.protos import deepvariant_pb2
  from third_party.nucleus.io import tfrecord
  from third_party.nucleus.testing import test_utils
  from third_party.nucleus.util import variant_utils
  from tensorflow.core.example import example_pb2
label_desc: Use image node_id 14 for calling the DataProviderTest obj's underlying member method code with example usage: obj.test_features_extraction_spec_for_mode() without return types