def testZerosShapeErrors(self):
    self.assertRaisesRegex(TypeError, 'Shapes must be 1D sequences of concrete values of integer type.*', (lambda : jnp.zeros(1.0)))
    self.assertRaisesRegex(TypeError, 'Shapes must be 1D sequences of concrete values of integer type.*\nIf using `jit`, try using `static_argnums` or applying `jit` to smaller subfunctions.', (lambda : api.jit(jnp.zeros)(2)))
def testZerosShapeErrors(self):
    self.assertRaisesRegex(TypeError, 'Shapes must be 1D sequences of concrete values of integer type.*', (lambda : jnp.zeros(1.0)))
    self.assertRaisesRegex(TypeError, 'Shapes must be 1D sequences of concrete values of integer type.*\nIf using `jit`, try using `static_argnums` or applying `jit` to smaller subfunctions.', (lambda : api.jit(jnp.zeros)(2)))