@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_{}'.format(jtu.format_shape_dtype_string(shape, dtype)), 'rng_factory': rng_factory, 'shape': shape, 'dtype': dtype} for shape in [(3,), (2, 3), (3, 2, 4)] for dtype in default_dtypes for rng_factory in [jtu.rand_default])))
def testFlipud(self, shape, dtype, rng_factory):
    rng = rng_factory()
    args_maker = self._GetArgsMaker(rng, [shape], [dtype])
    jnp_op = (lambda x: jnp.flipud(x))
    onp_op = (lambda x: onp.flipud(x))
    self._CheckAgainstNumpy(onp_op, jnp_op, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_op, args_maker, check_dtypes=True)
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_{}'.format(jtu.format_shape_dtype_string(shape, dtype)), 'rng_factory': rng_factory, 'shape': shape, 'dtype': dtype} for shape in [(3,), (2, 3), (3, 2, 4)] for dtype in default_dtypes for rng_factory in [jtu.rand_default])))
def testFlipud(self, shape, dtype, rng_factory):
    rng = rng_factory()
    args_maker = self._GetArgsMaker(rng, [shape], [dtype])
    jnp_op = (lambda x: jnp.flipud(x))
    onp_op = (lambda x: onp.flipud(x))
    self._CheckAgainstNumpy(onp_op, jnp_op, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_op, args_maker, check_dtypes=True)