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])