def test_fit_polynomial():
basis_function = Polynomial(degree=2)
data = np.array(([1, 1, 1], [2, 3, 4], [3, 3, 3]))
max_lag = 1
output = np.array([[4, 6, 8, 9, 12, 16], [9, 9, 9, 9, 9, 9]])
r = basis_function.fit(data=data, max_lag=max_lag)
assert_array_equal(output, r)
def test_fit_polynomial_predefined():
basis_function = Polynomial(degree=2)
data = np.array(([1, 1, 1], [2, 3, 4], [3, 3, 3]))
max_lag = 1
predefined_regressors = np.array([0, 2, 4])
output = np.array([[4, 8, 12], [9, 9, 9]])
r = basis_function.fit(
data=data, max_lag=max_lag, predefined_regressors=predefined_regressors
)
assert_array_equal(output, r)