code: def _verifySimpleShardingOutput(self, dataset, record_fn):
      next_element_fn = self._getNext(dataset)
      with self.cached_session():
          for f in range(
              self._shard_index, self._num_files, self._num_shards
          ):
              for r in range(self._num_records):
                  self.assertAllEqual(
                      record_fn(r, f), self.evaluate(next_element_fn())
                  )
          with self.assertRaises(errors.OutOfRangeError):
              self.evaluate(next_element_fn())
  
imports: import os
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import readers
  from tensorflow.python.data.util import structure
  from tensorflow.python.distribute import input_ops
  from tensorflow.python.eager import context
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import test_util
  from tensorflow.python.lib.io import python_io
  from tensorflow.python.ops import gen_dataset_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.util import compat
label_desc: Use image node_id 9 for calling the AutoShardDatasetTest obj's underlying member method code with example usage: obj._verifySimpleShardingOutput(dataset, record_fn) without return types