code: def test_splits_record_batch(self):
      feature1 = pa.array([1.0])
      feature2 = pa.array([2.0])
      feature3 = pa.array([3.0])
      record_batch = pa.RecordBatch.from_arrays(
          [feature1, feature2, feature3], ["a", "b", "c"]
      )
      sliced_record_batch = ("slice_key", record_batch)
  
      partitioner = mock.create_autospec(
          feature_partition_util.ColumnHasher(0)
      )
      partitioner.assign.side_effect = [99, 43, 99]
  
      # Verify we saw the right features.
      partitions = list(
          feature_partition_util.generate_feature_partitions(
              sliced_record_batch, partitioner, frozenset([])
          )
      )
      self.assertCountEqual(
          [mock.call("a"), mock.call("b"), mock.call("c")],
          partitioner.assign.call_args_list,
      )
  
      # Verify we got the right output slices.
      expected = {
          ("slice_key", 99): pa.RecordBatch.from_arrays(
              [feature1, feature3], ["a", "c"]
          ),
          ("slice_key", 43): pa.RecordBatch.from_arrays(
              [feature2], ["b"]
          ),
      }
      self.assertCountEqual(expected.keys(), [x[0] for x in partitions])
      for key, partitioned_record_batch in partitions:
          expected_batch = expected[key]
          test_util.make_arrow_record_batches_equal_fn(
              self, [expected_batch]
          )([partitioned_record_batch])
  
imports: from typing import Iterable, List, Tuple
  from unittest import mock
  from absl.testing import absltest
  from absl.testing import parameterized
  import apache_beam
  from apache_beam.testing import util
  import pyarrow
  from tensorflow_data_validation.utils import feature_partition_util
  from tensorflow_data_validation.utils import test_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 1 for calling the FeaturePartitionUtilTest obj's underlying member method code with example usage: obj.test_splits_record_batch() without return types