def test_poly1d_func(): p4 = np.array([3.0, 4, 5, 6]) pf = Poly1dFunction(np.poly1d(p4)) assert_equal(pf(2), 24 + 16 + 10 + 6) pf = Poly1dFunction.from_coeffs(p4) assert_equal(pf(2), 24 + 16 + 10 + 6)
def test_poly1d_func_ops(): p1 = Poly1dFunction.from_coeffs([5.0, 4]) p2 = Poly1dFunction.from_coeffs([2.0, 3, 4]) assert_equal((p1 * p2)(3), 19 * 31) assert_equal((p1 + p2)(3), 19 + 31)