@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_shape={}_n={}'.format(onp.dtype(dtype).name, n), 'dtype': dtype, 'n': n} for dtype in default_dtypes for n in list(range(4)))))
def testIdentity(self, n, dtype):
    onp_fun = (lambda : onp.identity(n, dtype))
    jnp_fun = (lambda : jnp.identity(n, dtype))
    args_maker = (lambda : [])
    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': '_shape={}_n={}'.format(onp.dtype(dtype).name, n), 'dtype': dtype, 'n': n} for dtype in default_dtypes for n in list(range(4)))))
def testIdentity(self, n, dtype):
    onp_fun = (lambda : onp.identity(n, dtype))
    jnp_fun = (lambda : jnp.identity(n, dtype))
    args_maker = (lambda : [])
    self._CheckAgainstNumpy(onp_fun, jnp_fun, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)