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)
