def testArange(self):
    self.assertAllClose(jnp.arange(77), onp.arange(77, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(2, 13), onp.arange(2, 13, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(4, 21, 9), onp.arange(4, 21, 9, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(53, 5, (- 3)), onp.arange(53, 5, (- 3), dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(77, dtype=float), onp.arange(77, dtype=float), check_dtypes=True)
    self.assertAllClose(jnp.arange(2, 13, dtype=int), onp.arange(2, 13, dtype=int), check_dtypes=True)
    self.assertAllClose(jnp.arange(0, 1, (- 0.5)), onp.arange(0, 1, (- 0.5), dtype=jnp.float_), check_dtypes=True)
    self.assertRaises(TypeError, (lambda : jnp.arange()))
    self.assertNotEqual(type(jnp.arange(77)), type(onp.arange(77)))
    self.assertEqual(type(jnp.arange(77)), type(lax.iota(onp.int32, 77)))
    self.assertNotEqual(type(jnp.arange(77, dtype=jnp.int32)), type(onp.arange(77, dtype=onp.int32)))
    self.assertEqual(type(jnp.arange(77, dtype=jnp.int32)), type(lax.iota(onp.int32, 77)))
    self.assertTrue(xla.is_device_constant(jnp.arange(77)))
    self.assertTrue(xla.is_device_constant(jnp.arange(77, dtype=jnp.int32)))
def testArange(self):
    self.assertAllClose(jnp.arange(77), onp.arange(77, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(2, 13), onp.arange(2, 13, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(4, 21, 9), onp.arange(4, 21, 9, dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(53, 5, (- 3)), onp.arange(53, 5, (- 3), dtype=jnp.int_), check_dtypes=True)
    self.assertAllClose(jnp.arange(77, dtype=float), onp.arange(77, dtype=float), check_dtypes=True)
    self.assertAllClose(jnp.arange(2, 13, dtype=int), onp.arange(2, 13, dtype=int), check_dtypes=True)
    self.assertAllClose(jnp.arange(0, 1, (- 0.5)), onp.arange(0, 1, (- 0.5), dtype=jnp.float_), check_dtypes=True)
    self.assertRaises(TypeError, (lambda : jnp.arange()))
    self.assertNotEqual(type(jnp.arange(77)), type(onp.arange(77)))
    self.assertEqual(type(jnp.arange(77)), type(lax.iota(onp.int32, 77)))
    self.assertNotEqual(type(jnp.arange(77, dtype=jnp.int32)), type(onp.arange(77, dtype=onp.int32)))
    self.assertEqual(type(jnp.arange(77, dtype=jnp.int32)), type(lax.iota(onp.int32, 77)))
    self.assertTrue(xla.is_device_constant(jnp.arange(77)))
    self.assertTrue(xla.is_device_constant(jnp.arange(77, dtype=jnp.int32)))