def testAstype(self):
    rng = onp.random.RandomState(0)
    args_maker = (lambda : [rng.randn(3, 4).astype('float32')])
    op = (lambda x: x.astype(jnp.int32))
    self._CheckAgainstNumpy(op, op, args_maker, check_dtypes=True)
    self._CompileAndCheck(op, args_maker, check_dtypes=True)
def testAstype(self):
    rng = onp.random.RandomState(0)
    args_maker = (lambda : [rng.randn(3, 4).astype('float32')])
    op = (lambda x: x.astype(jnp.int32))
    self._CheckAgainstNumpy(op, op, args_maker, check_dtypes=True)
    self._CompileAndCheck(op, args_maker, check_dtypes=True)