def testMsortManually(self):
    args_maker = (lambda : [onp.random.randint(50, size=(5, 5))])
    jnp_op = (lambda x: jnp.msort(x))
    onp_op = (lambda x: onp.msort(x))
    self._CheckAgainstNumpy(jnp_op, onp_op, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_op, args_maker, check_dtypes=True)
def testMsortManually(self):
    args_maker = (lambda : [onp.random.randint(50, size=(5, 5))])
    jnp_op = (lambda x: jnp.msort(x))
    onp_op = (lambda x: onp.msort(x))
    self._CheckAgainstNumpy(jnp_op, onp_op, args_maker, check_dtypes=True)
    self._CompileAndCheck(jnp_op, args_maker, check_dtypes=True)