def testIssue1233(self):
    '\n    Following numpy test suite from `test_repeat` at https://github.com/numpy/numpy/blob/master/numpy/core/tests/test_multiarray.py\n    '
    tol = 1e-05

    def test_single(m, args_maker, repeats, axis):
        lax_ans = jnp.repeat(m, repeats, axis)
        numpy_ans = onp.repeat(m, repeats, axis)
        self.assertAllClose(lax_ans, numpy_ans, check_dtypes=True, rtol=tol, atol=tol)
        jnp_fun = (lambda arg: jnp.repeat(arg, repeats=repeats, axis=axis))
        self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)
    m = jnp.array([1, 2, 3, 4, 5, 6])
    args_maker = (lambda : [m])
    for repeats in [2, [1, 3, 2, 1, 1, 2], [1, 3, 0, 1, 1, 2], [2], jnp.array([1, 3, 2, 1, 1, 2]), jnp.array([2])]:
        test_single(m, args_maker, repeats, None)
    m_rect = m.reshape((2, 3))
    args_maker = (lambda : [m_rect])
    for repeats in [2, [2, 1], [2], jnp.array([2, 1]), jnp.array([2])]:
        test_single(m_rect, args_maker, repeats, axis=0)
    for repeats in [2, [1, 3, 2], [2], jnp.array([1, 3, 2]), jnp.array([2])]:
        test_single(m_rect, args_maker, repeats, axis=1)
Following numpy test suite from `test_repeat` at https://github.com/numpy/numpy/blob/master/numpy/core/tests/test_multiarray.py
def testIssue1233(self):
    '\n    \n    '
    tol = 1e-05

    def test_single(m, args_maker, repeats, axis):
        lax_ans = jnp.repeat(m, repeats, axis)
        numpy_ans = onp.repeat(m, repeats, axis)
        self.assertAllClose(lax_ans, numpy_ans, check_dtypes=True, rtol=tol, atol=tol)
        jnp_fun = (lambda arg: jnp.repeat(arg, repeats=repeats, axis=axis))
        self._CompileAndCheck(jnp_fun, args_maker, check_dtypes=True)
    m = jnp.array([1, 2, 3, 4, 5, 6])
    args_maker = (lambda : [m])
    for repeats in [2, [1, 3, 2, 1, 1, 2], [1, 3, 0, 1, 1, 2], [2], jnp.array([1, 3, 2, 1, 1, 2]), jnp.array([2])]:
        test_single(m, args_maker, repeats, None)
    m_rect = m.reshape((2, 3))
    args_maker = (lambda : [m_rect])
    for repeats in [2, [2, 1], [2], jnp.array([2, 1]), jnp.array([2])]:
        test_single(m_rect, args_maker, repeats, axis=0)
    for repeats in [2, [1, 3, 2], [2], jnp.array([1, 3, 2]), jnp.array([2])]:
        test_single(m_rect, args_maker, repeats, axis=1)