@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_op={}_dtype={}'.format(op, pytype.__name__), 'pytype': pytype, 'dtype': dtype, 'op': op} for (pytype, dtype) in [(int, jnp.int_), (float, jnp.float_), (bool, jnp.bool_), (complex, jnp.complex_)] for op in ['atleast_1d', 'atleast_2d', 'atleast_3d'])))
def testAtLeastNdLiterals(self, pytype, dtype, op):
    onp_fun = (lambda arg: getattr(onp, op)(arg).astype(dtype))
    jnp_fun = (lambda arg: getattr(jnp, op)(arg))
    args_maker = (lambda : [pytype(2)])
    self._CheckAgainstNumpy(onp_fun, jnp_fun, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_op={}_dtype={}'.format(op, pytype.__name__), 'pytype': pytype, 'dtype': dtype, 'op': op} for (pytype, dtype) in [(int, jnp.int_), (float, jnp.float_), (bool, jnp.bool_), (complex, jnp.complex_)] for op in ['atleast_1d', 'atleast_2d', 'atleast_3d'])))
def testAtLeastNdLiterals(self, pytype, dtype, op):
    onp_fun = (lambda arg: getattr(onp, op)(arg).astype(dtype))
    jnp_fun = (lambda arg: getattr(jnp, op)(arg))
    args_maker = (lambda : [pytype(2)])
    self._CheckAgainstNumpy(onp_fun, jnp_fun, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)