diff options
Diffstat (limited to 'tests/test_lp.py')
| -rw-r--r-- | tests/test_lp.py | 22 |
1 files changed, 3 insertions, 19 deletions
diff --git a/tests/test_lp.py b/tests/test_lp.py index 8d80ba2..4537f46 100644 --- a/tests/test_lp.py +++ b/tests/test_lp.py @@ -1,28 +1,12 @@ -# Copyright (c) 2010 John Glover, National University of Ireland, Maynooth -# -# This program is free software; you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation; either version 2 of the License, or -# (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with this program; if not, write to the Free Software -# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - 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] |