code: def test_assign_buckets(
      self,
      x,
      x_spec,
      boundaries,
      boundaries_spec,
      side,
      expected_buckets,
      function_handler,
  ):
      @function_handler(input_signature=[x_spec, boundaries_spec])
      def _assign_buckets(x, boundaries):
          return tf_utils.assign_buckets(x, boundaries, side)
  
      buckets = _assign_buckets(x, boundaries)
      self.assertAllEqual(buckets, expected_buckets)
  
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 24 for calling the TFUtilsTest obj's underlying member method code with example usage: obj.test_assign_buckets(x, x_spec, boundaries, boundaries_spec, side, expected_buckets, function_handler) and returns: tf_utils