@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_shape={}'.format(jtu.format_shape_dtype_string(shape, dtype)), 'shape': shape, 'dtype': dtype} for shape in all_shapes for dtype in all_dtypes)))
def testWhereOneArgument(self, shape, dtype):
    rng = jtu.rand_some_zero()
    onp_fun = (lambda x: onp.where(x))
    onp_fun = jtu.ignore_warning(category=DeprecationWarning, message='Calling nonzero on 0d arrays.*')(onp_fun)
    jnp_fun = (lambda x: jnp.where(x))
    args_maker = (lambda : [rng(shape, dtype)])
    self._CheckAgainstNumpy(onp_fun, jnp_fun, args_maker, check_dtypes=False)
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_shape={}'.format(jtu.format_shape_dtype_string(shape, dtype)), 'shape': shape, 'dtype': dtype} for shape in all_shapes for dtype in all_dtypes)))
def testWhereOneArgument(self, shape, dtype):
    rng = jtu.rand_some_zero()
    onp_fun = (lambda x: onp.where(x))
    onp_fun = jtu.ignore_warning(category=DeprecationWarning, message='Calling nonzero on 0d arrays.*')(onp_fun)
    jnp_fun = (lambda x: jnp.where(x))
    args_maker = (lambda : [rng(shape, dtype)])
    self._CheckAgainstNumpy(onp_fun, jnp_fun, args_maker, check_dtypes=False)