blob: 4537f469f9a4f52e73b9b40b5cb3983f05b58195 (
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]
|