X-Git-Url: https://scm.cri.mines-paristech.fr/git/linpy.git/blobdiff_plain/ddefc755b0637b08a0a2788be3a0678b52942f5b..cc1d83eaadffc1d5de296e2ec2b401d04de70c41:/linpy/tests/test_polyhedra.py?ds=sidebyside diff --git a/linpy/tests/test_polyhedra.py b/linpy/tests/test_polyhedra.py index d0ebe57..60e1b33 100644 --- a/linpy/tests/test_polyhedra.py +++ b/linpy/tests/test_polyhedra.py @@ -15,11 +15,10 @@ # You should have received a copy of the GNU General Public License # along with LinPy. If not, see . -import functools import unittest from ..linexprs import symbols -from ..polyhedra import * +from ..polyhedra import Empty, Polyhedron, Universe from .libhelper import requires_sympy @@ -37,11 +36,13 @@ class TestPolyhedron(unittest.TestCase): def test_repr(self): self.assertEqual(repr(self.square), - "And(0 <= x, x <= 1, 0 <= y, y <= 1)") + 'And(0 <= x, x <= 1, 0 <= y, y <= 1)') def test_fromstring(self): - self.assertEqual(Polyhedron.fromstring('{x >= 0, -x + 1 >= 0, ' - 'y >= 0, -y + 1 >= 0}'), self.square) + self.assertEqual( + Polyhedron.fromstring( + '{x >= 0, -x + 1 >= 0, y >= 0, -y + 1 >= 0}'), + self.square) def test_isempty(self): self.assertFalse(self.square.isempty()) @@ -53,14 +54,17 @@ class TestPolyhedron(unittest.TestCase): def test_fromsympy(self): import sympy sp_x, sp_y = sympy.symbols('x y') - self.assertEqual(Polyhedron.fromsympy((sp_x >= 0) & (sp_x <= 1) & - (sp_y >= 0) & (sp_y <= 1)), self.square) + self.assertEqual( + Polyhedron.fromsympy( + (sp_x >= 0) & (sp_x <= 1) & (sp_y >= 0) & (sp_y <= 1)), + self.square) @requires_sympy def test_tosympy(self): import sympy sp_x, sp_y = sympy.symbols('x y') - self.assertEqual(self.square.tosympy(), + self.assertEqual( + self.square.tosympy(), sympy.And(-sp_x + 1 >= 0, -sp_y + 1 >= 0, sp_x >= 0, sp_y >= 0))