code: def test_stats_impl(
      self,
      record_batches,
      options,
      expected_result_proto_text,
      expected_shards=1,
      schema=None,
  ):
      expected_result = text_format.Parse(
          expected_result_proto_text,
          statistics_pb2.DatasetFeatureStatisticsList(),
      )
      if schema is not None:
          options.schema = schema
      with beam.Pipeline() as p:
          result = (
              p
              | beam.Create(record_batches, reshuffle=False)
              | stats_impl.GenerateStatisticsImpl(options)
          )
          if expected_shards > 1:
              merge_fn = _merge_shards
          else:
              merge_fn = None
          util.assert_that(
              result,
              test_util.make_dataset_feature_stats_list_proto_equal_fn(
                  self,
                  expected_result,
                  expected_result_len=expected_shards,
                  expected_result_merge_fn=merge_fn,
                  check_histograms=False,
              ),
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import copy
  import sys
  from typing import Iterable
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import apache_beam
  from apache_beam.testing import util
  import numpy
  import pyarrow
  from tensorflow_data_validation.statistics import stats_impl
  from tensorflow_data_validation.statistics import stats_options
  from tensorflow_data_validation.statistics.generators import cross_feature_stats_generator
  from tensorflow_data_validation.statistics.generators import stats_generator
  from tensorflow_data_validation.utils import slicing_util
  from tensorflow_data_validation.utils import test_util
  from tfx_bsl.arrow import array_util
  from tfx_bsl.arrow import table_util
  from tfx_bsl.public.proto import slicing_spec_pb2
  from tfx_bsl.statistics import merge_util
  from google.protobuf import text_format
  from tensorflow.python.util.protobuf import compare
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 1 for calling the StatsImplTest obj's underlying member method code with example usage: obj.test_stats_impl(record_batches, options, expected_result_proto_text, expected_shards, schema) without return types