blob: ee13b51bd30dfaa158e6ef96e798a1a5e58cb99e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
from simpl import lp
import numpy as np
class TestLP(object):
def test_predict(self):
"""test_predict"""
coefs = np.array([1,2,3,4,5])
test_signal = np.ones(5)
predictions = lp.predict(test_signal, coefs, 2)
assert predictions[0] == -sum(coefs)
assert predictions[1] == -sum(coefs[1:])-predictions[0]
|