Docstrings
[linpy.git] / linpy / tests / libhelper.py
1 # Copyright 2014 MINES ParisTech
2 #
3 # This file is part of LinPy.
4 #
5 # LinPy is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # LinPy is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with LinPy. If not, see <http://www.gnu.org/licenses/>.
17
18 import functools
19 import unittest
20
21 try:
22
23 import sympy
24
25 def requires_sympy(func):
26 @functools.wraps(func)
27 def wrapper(self):
28 return func(self)
29 return wrapper
30
31 except ImportError:
32
33 def requires_sympy(func):
34 @functools.wraps(func)
35 def wrapper(self):
36 raise unittest.SkipTest('SymPy is not available')
37 return wrapper