code: def test_axis_set_too_large_raises(self):
      with self.assertRaisesRegexp(ValueError, "Since dims"):
          interpolation.batch_interp_rectilinear_nd_grid(
              x=[[1.0]],
              x_grid_points=([0, 1, 2.0],),
              y_ref=[1.0, 2.0, 3.0],
              axis=3,
          )
  
imports: import numpy
  from scipy import interpolate
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.math import interpolation
label_desc: Use image node_id 2 for calling the BatchInterpRectilinearNDGridTest obj's underlying member method code with example usage: obj.test_axis_set_too_large_raises() without return types