def test_addition(self): pp1 = PiecewisePolynomial([1, 2, 3]) pp2 = pp1 + pp1 self.assertTrue((pp2.value(0) == 2 * pp1.value(0)).all())
def test_piecewise_polynomial_constant_constructor(self): x = np.array([[1.], [4.]]) pp = PiecewisePolynomial(x) self.assertEqual(pp.rows(), 2) self.assertEqual(pp.cols(), 1) np.testing.assert_equal(x, pp.value(11.))