def testNonArrayErrorMessage(self):
    x = [1.0, 2.0]
    y = onp.array([3.0, 4.0])

    def g(x, y):
        return jnp.add(x, y)

    def f(x, y):
        return jnp.dot(x, y)
    self.assertRaises(TypeError, (lambda : g(x, y)))
    self.assertRaises(TypeError, (lambda : f(x, y)))
    self.assertRaises(TypeError, (lambda : api.jit(g)(x, y)))
    self.assertRaises(TypeError, (lambda : api.jit(f)(x, y)))
def testNonArrayErrorMessage(self):
    x = [1.0, 2.0]
    y = onp.array([3.0, 4.0])

    def g(x, y):
        return jnp.add(x, y)

    def f(x, y):
        return jnp.dot(x, y)
    self.assertRaises(TypeError, (lambda : g(x, y)))
    self.assertRaises(TypeError, (lambda : f(x, y)))
    self.assertRaises(TypeError, (lambda : api.jit(g)(x, y)))
    self.assertRaises(TypeError, (lambda : api.jit(f)(x, y)))