code: def test_reduce_batch_count_per_key(
      self,
      key,
      spec,
      expected_key_vocab,
      expected_count,
      function_handler,
  ):
      @function_handler(input_signature=[spec])
      def _reduce_batch_count_per_key(key):
          return tf_utils.reduce_batch_count_per_key(key)
  
      key_vocab, key_counts = _reduce_batch_count_per_key(key)
  
      self.assertAllEqual(key_vocab, expected_key_vocab)
      self.assertAllEqual(key_counts, expected_count)
  
imports: import os
  import numpy
  from packaging import version
  import tensorflow
  from tensorflow_transform import analyzers
  from tensorflow_transform import annotators
  from tensorflow_transform import tf_utils
  from tensorflow_transform import test_case
  from tensorflow.python.framework import composite_tensor
label_desc: Use image node_id 22 for calling the TFUtilsTest obj's underlying member method code with example usage: obj.test_reduce_batch_count_per_key(key, spec, expected_key_vocab, expected_count, function_handler) and returns: tf_utils