code: def test_reading_sharded_dataset(self, compressed_inputs, use_tpu):
      golden_dataset = make_golden_dataset(
          compressed_inputs, use_tpu=use_tpu
      )
      n_shards = 3
      sharded_path = test_utils.test_tmpfile(
          "sharded@{}".format(n_shards)
      )
      tfrecord.write_tfrecords(
          tfrecord.read_tfrecords(golden_dataset.input_file_spec),
          sharded_path,
      )
  
      config_file = _test_dataset_config(
          "test_sharded.pbtxt",
          name="sharded_test",
          tfrecord_path=sharded_path,
          num_examples=golden_dataset.num_examples,
      )
  
      self.assertTfDataSetExamplesMatchExpected(
          data_providers.get_input_fn_from_dataset(
              config_file, mode=tf_estimator.ModeKeys.EVAL
          ),
          golden_dataset,
          # workaround_list_files is needed because wildcards, and so sharded
          # files, are nondeterministicly ordered (for now).
          workaround_list_files=True,
      )
  
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 8 for calling the DataProviderTest obj's underlying member method code with example usage: obj.test_reading_sharded_dataset(compressed_inputs, use_tpu) without return types