@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': jtu.format_test_name_suffix('ix_', shapes, dtypes), 'rng_factory': jtu.rand_default, 'shapes': shapes, 'dtypes': dtypes} for (shapes, dtypes) in (((), ()), (((7,),), (onp.int32,)), (((3,), (4,)), (onp.int32, onp.int32)), (((3,), (1,), (4,)), (onp.int32, onp.int32, onp.int32))))))
def testIx_(self, rng_factory, shapes, dtypes):
    rng = rng_factory()
    args_maker = (lambda : [rng(shape, dtype) for (shape, dtype) in zip(shapes, dtypes)])
    self._CheckAgainstNumpy(onp.ix_, jnp.ix_, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp.ix_, args_maker, check_dtypes=True)
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': jtu.format_test_name_suffix('ix_', shapes, dtypes), 'rng_factory': jtu.rand_default, 'shapes': shapes, 'dtypes': dtypes} for (shapes, dtypes) in (((), ()), (((7,),), (onp.int32,)), (((3,), (4,)), (onp.int32, onp.int32)), (((3,), (1,), (4,)), (onp.int32, onp.int32, onp.int32))))))
def testIx_(self, rng_factory, shapes, dtypes):
    rng = rng_factory()
    args_maker = (lambda : [rng(shape, dtype) for (shape, dtype) in zip(shapes, dtypes)])
    self._CheckAgainstNumpy(onp.ix_, jnp.ix_, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp.ix_, args_maker, check_dtypes=True)