code: def testInvalidAxisShape(self):
      # The axis should be a scalar or 1-D, checked in shape function.
      with self.assertRaisesRegex(
          ValueError, "Shape must be at most rank 1 but is rank 2"
      ):
          manip_ops.roll([[1, 2], [3, 4]], 1, [[0, 1]])
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradient_checker
  from tensorflow.python.ops import manip_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 12 for calling the RollTest obj's underlying member method code with example usage: obj.testInvalidAxisShape() without return types