diff options
Diffstat (limited to 'tests/test_lp.py')
| -rw-r--r-- | tests/test_lp.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_lp.py b/tests/test_lp.py index ee13b51..4537f46 100644 --- a/tests/test_lp.py +++ b/tests/test_lp.py @@ -1,12 +1,12 @@ from simpl import lp import numpy as np -class TestLP(object): +class TestLP(object): def test_predict(self): """test_predict""" - coefs = np.array([1,2,3,4,5]) + 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] + assert predictions[1] == -sum(coefs[1:]) - predictions[0] |