try:
import sympy
- def _with_sympy(func):
+ def _requires_sympy(func):
@functools.wraps(func)
def wrapper(self):
return func(self)
return wrapper
except ImportError:
- def _with_sympy(func):
+ def _requires_sympy(func):
@functools.wraps(func)
def wrapper(self):
raise unittest.SkipTest('SymPy is not available')
self.assertEqual((self.x + self.y/2 + self.z/3)._toint(),
6*self.x + 3*self.y + 2*self.z)
- @_with_sympy
+ @_requires_sympy
def test_fromsympy(self):
sp_x, sp_y = sympy.symbols('x y')
self.assertEqual(Expression.fromsympy(sp_x), self.x)
with self.assertRaises(ValueError):
Expression.fromsympy(sp_x*sp_y)
- @_with_sympy
+ @_requires_sympy
def test_tosympy(self):
sp_x, sp_y = sympy.symbols('x y')
self.assertEqual(self.x.tosympy(), sp_x)
self.one = Constant(1)
self.pi = Constant(Fraction(22, 7))
- @_with_sympy
+ @_requires_sympy
def test_fromsympy(self):
self.assertEqual(Constant.fromsympy(sympy.Rational(22, 7)), self.pi)
with self.assertRaises(TypeError):
self.assertListEqual(list(symbols('x,y')), [self.x, self.y])
self.assertListEqual(list(symbols(['x', 'y'])), [self.x, self.y])
- @_with_sympy
+ @_requires_sympy
def test_fromsympy(self):
sp_x = sympy.Symbol('x')
self.assertEqual(Symbol.fromsympy(sp_x), self.x)
self.assertFalse(self.square.isuniverse())
@unittest.expectedFailure
- @_with_sympy
+ @_requires_sympy
def test_fromsympy(self):
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)
- @_with_sympy
+ @_requires_sympy
def test_tosympy(self):
sp_x, sp_y = sympy.symbols('x y')
self.assertEqual(self.square.tosympy(),