code: def testReadOneEpoch(self):
      # Basic test: read from file 0.
      dataset = self._dataset_factory(self._filenames[0])
      self.assertDatasetProduces(
          dataset,
          expected_output=[
              self._record(0, i) for i in range(self._num_records)
          ],
      )
  
      # Basic test: read from file 1.
      dataset = self._dataset_factory(self._filenames[1])
      self.assertDatasetProduces(
          dataset,
          expected_output=[
              self._record(1, i) for i in range(self._num_records)
          ],
      )
  
      # Basic test: read from both files.
      dataset = self._dataset_factory(self._filenames)
      expected_output = []
      for j in range(self._num_files):
          expected_output.extend(
              [self._record(j, i) for i in range(self._num_records)]
          )
      self.assertDatasetProduces(
          dataset, expected_output=expected_output
      )
  
imports: import gzip
  import os
  import pathlib
  import zlib
  from absl.testing import parameterized
  from tensorflow.python.data.kernel_tests import checkpoint_test_base
  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.ops import options
  from tensorflow.python.data.ops import readers
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import constant_op
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the TFRecordDatasetTest obj's underlying member method code with example usage: obj.testReadOneEpoch() without return types