pypol/tests/libhelper.py

   1 import functools
   2 import unittest
   3
   4 try:
   5
   6     import sympy
   7
   8     def requires_sympy(func):
   9         @functools.wraps(func)
  10         def wrapper(self):
  11             return func(self)
  12         return wrapper
  13
  14 except ImportError:
  15
  16     def requires_sympy(func):
  17         @functools.wraps(func)
  18         def wrapper(self):
  19             raise unittest.SkipTest('SymPy is not available')
  20         return wrapper