code: def test_load_stats_tfrecord(self):
      stats = text_format.Parse(
          """
        datasets { name: 'abc' }
      """,
          statistics_pb2.DatasetFeatureStatisticsList(),
      )
      stats_path = os.path.join(FLAGS.test_tmpdir, "stats.tfrecord")
      with tf.io.TFRecordWriter(stats_path) as writer:
          writer.write(stats.SerializeToString())
      self.assertEqual(
          stats, stats_util.load_stats_tfrecord(input_path=stats_path)
      )
      self.assertEqual(
          stats, stats_util.load_statistics(input_path=stats_path)
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from absl import flags
  from absl.testing import absltest
  import numpy
  import tensorflow
  from tensorflow_data_validation import types
  from tensorflow_data_validation.utils import artifacts_io_impl
  from tensorflow_data_validation.utils import stats_util
  from google.protobuf import text_format
  from tensorflow.python.util.protobuf import compare
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 8 for calling the StatsUtilTest obj's underlying member method code with example usage: obj.test_load_stats_tfrecord() without return types