def testArangeOnFloats(self):
    expected = onp.arange(0.0, 1.0, 0.1, dtype=jnp.float_)
    ans = jnp.arange(0.0, 1.0, 0.1)
    self.assertAllClose(expected, ans, check_dtypes=True)
def testArangeOnFloats(self):
    expected = onp.arange(0.0, 1.0, 0.1, dtype=jnp.float_)
    ans = jnp.arange(0.0, 1.0, 0.1)
    self.assertAllClose(expected, ans, check_dtypes=True)