code: def testPerBatchBudgetTrimmer(
      self, max_seq_length, segments, expected, axis=-1, descr=None
  ):
      max_seq_length = constant_op.constant(max_seq_length)
      trimmer = trimmer_ops.RoundRobinTrimmer(max_seq_length, axis=axis)
      segments = [ragged_factory_ops.constant(seg) for seg in segments]
      expected = [ragged_factory_ops.constant(exp) for exp in expected]
      actual = trimmer.trim(segments)
      for expected_seg, actual_seg in zip(expected, actual):
          self.assertAllEqual(expected_seg, actual_seg)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  import tensorflow_text
  from tensorflow.lite.python import interpreter
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops.ragged import ragged_factory_ops
  from tensorflow.python.platform import test
  from tensorflow_text.python.ops import trimmer_ops
label_desc: Use image node_id 2 for calling the RoundRobinTrimmerOpsTest obj's underlying member method code with example usage: obj.testPerBatchBudgetTrimmer(max_seq_length, segments, expected, axis, descr) without return types