From 0ecf55c21718ce97384f0c44d797549fce29db12 Mon Sep 17 00:00:00 2001 From: John Glover Date: Fri, 24 Jun 2011 18:21:52 +0100 Subject: Update test file names --- tests/test_lp.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 tests/test_lp.py (limited to 'tests/test_lp.py') diff --git a/tests/test_lp.py b/tests/test_lp.py new file mode 100644 index 0000000..8d80ba2 --- /dev/null +++ b/tests/test_lp.py @@ -0,0 +1,28 @@ +# 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): + + 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] -- cgit v1.2.3