code: def testOp(self, op_name, rng_factory, shapes, dtype, bdims, tol):
      if dtype == np.float64 or any(len(shape) > 2 for shape in shapes):
          self.skipTest(
              "Skipping big tests under sanitizers due to slowdown."
          )
  
      rng = rng_factory(self.rng())
      op = getattr(lax, op_name)
      self._CheckBatching(
          op,
          10,
          bdims,
          shapes,
          [dtype] * len(shapes),
          rng,
          atol=tol,
          rtol=tol,
      )
  
file_path: jax/tests/lax_vmap_op_test.py
imports: import functools
  import itertools
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import jax
  from jax import lax
  from jax._src import test_util
  from jax._src.internal_test_util import lax_test_util
  from jax._src import util
  from jax import config
label_desc: Use image node_id 2 for calling the LaxVmapOpTest obj's underlying member method code with example usage: obj.testOp(op_name, rng_factory, shapes, dtype, bdims, tol) without return types