def test_Lambdify():
ns = NameSpace(2)
x = ns.Symbol('x')
y = ns.Symbol('y')
z = ns.Symbol('z')
u = ns.Symbol('u')
v = ns.Symbol('v')
w = ns.Symbol('w')
lmb0 = ns.Lambdify((x, y, z, u, v, w), [x, y, z, u, v, w])
out0 = lmb0(np.array([2., 3, 4, 5, 6, 7]))
ref0 = [2., 3, 4, 5, 6, 7]
assert np.allclose(out0.flatten(), ref0)
lmb3 = ns.Lambdify((z, y, x), [x, y, z])
out3 = lmb3(np.array([2., 3, 4]))
ref3 = [4, 3, 2]
assert np.allclose(out3.flatten(), ref3)
lmb1 = ns.Lambdify((x, y), [x + y, x - y, x * y, 13 * x + 2 * y])
out1 = lmb1(np.array([2.0, 3.0]))
ref1 = [5, -1, 6, 13 * 2 + 2 * 3]
assert np.allclose(out1.flatten(), ref1)
lmb2 = ns.Lambdify((y, x), [x + y, x - y, x * y])
out2 = lmb2(np.array([2.0, 3.0]))
ref2 = [5, 1, 6]
assert np.allclose(out2.flatten(), ref2)
def test_Dummy():
ns = NameSpace()
d = ns.Dummy()
lmb = ns.Lambdify([d], [3 * d, 8])
assert np.allclose(lmb(np.array([7.0])), [21, 8])
