PEP 8
[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 importlib
20 import unittest
21
22 try:
23
24 importlib.import_module('sympy')
25
26 def requires_sympy(func):
27 @functools.wraps(func)
28 def wrapper(self):
29 return func(self)
30 return wrapper
31
32 except ImportError:
33
34 def requires_sympy(func):
35 @functools.wraps(func)
36 def wrapper(self):
37 raise unittest.SkipTest('SymPy is not available')
38 return wrapper