code: def testRaggedExpandDims(
      self,
      rt_input,
      axis,
      expected,
      ragged_rank=None,
      expected_shape=None,
  ):
      rt = ragged_factory_ops.constant(
          rt_input, ragged_rank=ragged_rank
      )
      expanded = ragged_array_ops.expand_dims(rt, axis=axis)
      self.assertEqual(expanded.shape.ndims, rt.shape.ndims + 1)
      if expected_shape is not None:
          self.assertEqual(expanded.shape.as_list(), expected_shape)
  
      self.assertAllEqual(expanded, expected)
  
file_path: tensorflow-upstream/tensorflow/python/ops/ragged/ragged_expand_dims_op_test.py
imports: from absl.testing import parameterized
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops.ragged import ragged_array_ops
  from tensorflow.python.ops.ragged import ragged_factory_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 1 for calling the RaggedExpandDimsOpTest obj's underlying member method code with example usage: obj.testRaggedExpandDims(rt_input, axis, expected, ragged_rank, expected_shape) without return types