def testWhereScalarPromotion(self):
    x = jnp.where(jnp.array([True, False]), 3, jnp.ones((2,), dtype=jnp.float32))
    self.assertEqual(x.dtype, onp.dtype(onp.float32))
def testWhereScalarPromotion(self):
    x = jnp.where(jnp.array([True, False]), 3, jnp.ones((2,), dtype=jnp.float32))
    self.assertEqual(x.dtype, onp.dtype(onp.float32))