code: def testShuffle(
      self, batch_size, num_epochs, num_parallel_reads, seed
  ):
      def dataset_fn():
          return readers.make_tf_record_dataset(
              file_pattern=self._filenames,
              num_epochs=num_epochs,
              batch_size=batch_size,
              num_parallel_reads=num_parallel_reads,
              shuffle=True,
              shuffle_seed=seed,
          )
  
      next_element = self.getNext(dataset_fn())
      first_batches = []
      try:
          while True:
              first_batches.append(self.evaluate(next_element()))
      except errors.OutOfRangeError:
          pass
  
      next_element = self.getNext(dataset_fn())
      second_batches = []
      try:
          while True:
              second_batches.append(self.evaluate(next_element()))
      except errors.OutOfRangeError:
          pass
  
      self.assertEqual(len(first_batches), len(second_batches))
      if seed is not None:
          # if you set a seed, should get the same results
          for i in range(len(first_batches)):
              self.assertAllEqual(first_batches[i], second_batches[i])
  
      expected = []
      for f in range(self._num_files):
          for r in range(self._num_records):
              expected.extend([self._record(f, r)] * num_epochs)
  
      for batches in (first_batches, second_batches):
          actual = []
          for b in batches:
              actual.extend(b)
          self.assertAllEqual(sorted(expected), sorted(actual))
  
file_path: tensorflow/tensorflow/python/data/experimental/kernel_tests/make_tf_record_dataset_test.py
imports: from absl.testing import parameterized
  from tensorflow.python.data.experimental.ops import readers
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.kernel_tests import tf_record_test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.util import nest
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import errors
  from tensorflow.python.ops import string_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the MakeTFRecordDatasetTest obj's underlying member method code with example usage: obj.testShuffle(batch_size, num_epochs, num_parallel_reads, seed) and returns: readers