@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_shapes={}_dtype={}_indexing={}_sparse={}'.format(shapes, dtype, indexing, sparse), 'shapes': shapes, 'dtype': dtype, 'indexing': indexing, 'sparse': sparse, 'rng_factory': rng_factory} for shapes in [(), (5,), (5, 3)] for dtype in number_dtypes for indexing in ['xy', 'ij'] for sparse in [True, False] for rng_factory in [jtu.rand_default])))
def testMeshGrid(self, shapes, dtype, indexing, sparse, rng_factory):
    rng = rng_factory()
    args_maker = self._GetArgsMaker(rng, [(x,) for x in shapes], ([dtype] * len(shapes)))
    onp_fun = partial(onp.meshgrid, indexing=indexing, sparse=sparse)
    jnp_fun = partial(jnp.meshgrid, indexing=indexing, sparse=sparse)
    self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_shapes={}_dtype={}_indexing={}_sparse={}'.format(shapes, dtype, indexing, sparse), 'shapes': shapes, 'dtype': dtype, 'indexing': indexing, 'sparse': sparse, 'rng_factory': rng_factory} for shapes in [(), (5,), (5, 3)] for dtype in number_dtypes for indexing in ['xy', 'ij'] for sparse in [True, False] for rng_factory in [jtu.rand_default])))
def testMeshGrid(self, shapes, dtype, indexing, sparse, rng_factory):
    rng = rng_factory()
    args_maker = self._GetArgsMaker(rng, [(x,) for x in shapes], ([dtype] * len(shapes)))
    onp_fun = partial(onp.meshgrid, indexing=indexing, sparse=sparse)
    jnp_fun = partial(jnp.meshgrid, indexing=indexing, sparse=sparse)
    self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)