X-Git-Url: https://scm.cri.ensmp.fr/git/linpy.git/blobdiff_plain/1d494bb187b70135df721c13306d7f26fdf33f50..d9ce6feb2d36e40e83326744f1d4ff3890d1874f:/pypol/tests/test_polyhedra.py diff --git a/pypol/tests/test_polyhedra.py b/pypol/tests/test_polyhedra.py index c74e25f..e813609 100644 --- a/pypol/tests/test_polyhedra.py +++ b/pypol/tests/test_polyhedra.py @@ -3,21 +3,7 @@ import unittest from ..linexprs import symbols from ..polyhedra import * - - -try: - import sympy - def _requires_sympy(func): - @functools.wraps(func) - def wrapper(self): - return func(self) - return wrapper -except ImportError: - def _requires_sympy(func): - @functools.wraps(func) - def wrapper(self): - raise unittest.SkipTest('SymPy is not available') - return wrapper +from .libhelper import requires_sympy class TestPolyhedron(unittest.TestCase): @@ -27,7 +13,7 @@ class TestPolyhedron(unittest.TestCase): self.square = Polyhedron(inequalities=[x, 1 - x, y, 1 - y]) def test_symbols(self): - self.assertCountEqual(self.square.symbols, ['x', 'y']) + self.assertTupleEqual(self.square.symbols, symbols('x y')) def test_dimension(self): self.assertEqual(self.square.dimension, 2) @@ -50,14 +36,16 @@ class TestPolyhedron(unittest.TestCase): def test_isuniverse(self): self.assertFalse(self.square.isuniverse()) - @_requires_sympy + @requires_sympy 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) - @_requires_sympy + @requires_sympy def test_tosympy(self): + import sympy sp_x, sp_y = sympy.symbols('x y') self.assertEqual(self.square.tosympy(), sympy.And(-sp_x + 1 >= 0, -sp_y + 1 >= 0, sp_x >= 0, sp_y >= 0))