from fractions import Fraction
from ..linexprs 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 TestExpression(unittest.TestCase):
self.assertEqual(self.pi.subs('x', 3), self.pi)
self.assertEqual(self.expr.subs('x', -3), -2 * self.y)
self.assertEqual(self.expr.subs([('x', self.y), ('y', self.x)]), 3 - self.x)
- self.assertEqual(self.expr.subs({'x': self.y, 'y': self.x}), 3 - self.x)
- self.assertEqual(self.expr.subs({self.x: self.z, self.y: self.z}), -self.z + 3)
+ self.assertEqual(self.expr.subs({'x': self.z, 'y': self.z}), 3 - self.z)
+ self.assertEqual(self.expr.subs({self.x: self.z, self.y: self.z}), 3 - self.z)
def test_fromstring(self):
self.assertEqual(Expression.fromstring('x'), self.x)
self.assertEqual(repr(self.x + self.one), "Expression('x + 1')")
self.assertEqual(repr(self.expr), "Expression('x - 2*y + 3')")
- @_requires_sympy
+ @requires_sympy
def test_fromsympy(self):
+ import sympy
sp_x, sp_y = sympy.symbols('x y')
self.assertEqual(Expression.fromsympy(sp_x), self.x)
self.assertEqual(Expression.fromsympy(sympy.Rational(22, 7)), self.pi)
with self.assertRaises(ValueError):
Expression.fromsympy(sp_x*sp_y)
- @_requires_sympy
+ @requires_sympy
def test_tosympy(self):
+ import sympy
sp_x, sp_y = sympy.symbols('x y')
self.assertEqual(self.x.tosympy(), sp_x)
self.assertEqual(self.pi.tosympy(), sympy.Rational(22, 7))
def test_repr(self):
self.assertEqual(repr(self.x), "Symbol('x')")
- @_requires_sympy
+ @requires_sympy
def test_fromsympy(self):
+ import sympy
sp_x = sympy.Symbol('x')
self.assertEqual(Symbol.fromsympy(sp_x), self.x)
with self.assertRaises(TypeError):
self.assertEqual(repr(self.one), 'Constant(1)')
self.assertEqual(repr(self.pi), 'Constant(22, 7)')
- @_requires_sympy
+ @requires_sympy
def test_fromsympy(self):
+ import sympy
self.assertEqual(Constant.fromsympy(sympy.Rational(22, 7)), self.pi)
with self.assertRaises(TypeError):
Constant.fromsympy(sympy.Symbol('x'))
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):
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))