def testOnesWithInvalidShape(self):
    with self.assertRaises(TypeError):
        jnp.ones(((- 1), 1))
def testOnesWithInvalidShape(self):
    with self.assertRaises(TypeError):
        jnp.ones(((- 1), 1))