code: def testTFRecordDataset(self):
      all_contents = []
      for i in range(_NUM_FILES):
          filename = os.path.join(
              self.get_temp_dir(), "tf_record.%d" % i
          )
          writer = python_io.TFRecordWriter(filename)
          for j in range(_NUM_ENTRIES):
              record = compat.as_bytes("Record %d of file %d" % (j, i))
              writer.write(record)
              all_contents.append(record)
          writer.close()
  
      dataset = datasets.StreamingFilesDataset(
          os.path.join(self.get_temp_dir(), "tf_record*"),
          filetype="tfrecord",
      )
  
      with ops.device(self._worker_device):
          iterator = dataset_ops.make_initializable_iterator(dataset)
      self._sess.run(iterator.initializer)
      get_next = iterator.get_next()
  
      retrieved_values = []
      for _ in range(4 * len(all_contents)):
          retrieved_values.append(
              compat.as_bytes(self._sess.run(get_next))
          )
  
      self.assertEqual(set(all_contents), set(retrieved_values))
  
imports: import os
  from tensorflow.core.protobuf import cluster_pb2
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.python.client import session
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import readers
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.lib.io import python_io
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import datasets
  from tensorflow.python.training import server_lib
  from tensorflow.python.util import compat
label_desc: Use image node_id 3 for calling the DatasetsTest obj's underlying member method code with example usage: obj.testTFRecordDataset() without return types