def testOperatorRound(self):
    self.assertAllClose(round(onp.float32(7.532), 1), round(jnp.float32(7.5), 1), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234), 2), round(jnp.float32(1.234), 2), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234)), round(jnp.float32(1.234)), check_dtypes=False)
    self.assertAllClose(round(onp.float32(7.532), 1), round(jnp.array(7.5, jnp.float32), 1), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234), 2), round(jnp.array(1.234, jnp.float32), 2), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234)), round(jnp.array(1.234, jnp.float32)), check_dtypes=False)
def testOperatorRound(self):
    self.assertAllClose(round(onp.float32(7.532), 1), round(jnp.float32(7.5), 1), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234), 2), round(jnp.float32(1.234), 2), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234)), round(jnp.float32(1.234)), check_dtypes=False)
    self.assertAllClose(round(onp.float32(7.532), 1), round(jnp.array(7.5, jnp.float32), 1), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234), 2), round(jnp.array(1.234, jnp.float32), 2), check_dtypes=True)
    self.assertAllClose(round(onp.float32(1.234)), round(jnp.array(1.234, jnp.float32)), check_dtypes=False)