1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# Define decorators that can be used for pytest tests
import os
import zlib
import tempfile
from contextlib import contextmanager
from distutils.version import LooseVersion
import pytest
def make_skipper(module, label=None, version=None):
label = label or module
try:
mod = __import__(module)
if version:
assert LooseVersion(mod.__version__) >= LooseVersion(version)
installed = True
except (ImportError, AssertionError):
installed = False
return installed, pytest.mark.skipif(str(not installed), reason='Requires %s' % label)
ASTROPY_INSTALLED, requires_astropy = make_skipper('astropy',
label='Astropy')
MATPLOTLIB_GE_22, requires_matplotlib_ge_22 = make_skipper('matplotlib', version='2.2')
ASTRODENDRO_INSTALLED, requires_astrodendro = make_skipper('astrodendro')
SCIPY_INSTALLED, requires_scipy = make_skipper('scipy',
label='SciPy')
PIL_INSTALLED, requires_pil = make_skipper('PIL', label='PIL')
SKIMAGE_INSTALLED, requires_skimage = make_skipper('skimage',
label='scikit-image')
XLRD_INSTALLED, requires_xlrd = make_skipper('xlrd')
PYAVM_INSTALLED, requires_pyavm = make_skipper('pyavm')
PLOTLY_INSTALLED, requires_plotly = make_skipper('plotly')
IPYTHON_INSTALLED, requires_ipython = make_skipper('IPython')
requires_pil_or_skimage = pytest.mark.skipif(str(not SKIMAGE_INSTALLED and not PIL_INSTALLED),
reason='Requires PIL or scikit-image')
PLOTLY_INSTALLED, requires_plotly = make_skipper('plotly')
H5PY_INSTALLED, requires_h5py = make_skipper('h5py')
PYQT5_INSTALLED, requires_pyqt5 = make_skipper('PyQt5')
PYSIDE2_INSTALLED, requires_pyside2 = make_skipper('PySide2')
HYPOTHESIS_INSTALLED, requires_hypothesis = make_skipper('hypothesis')
QT_INSTALLED = PYQT5_INSTALLED or PYSIDE2_INSTALLED
SPECTRAL_CUBE_INSTALLED, requires_spectral_cube = make_skipper('spectral_cube',
label='spectral-cube')
requires_qt = pytest.mark.skipif(str(not QT_INSTALLED),
reason='An installation of Qt is required')
@contextmanager
def make_file(contents, suffix, decompress=False):
"""Context manager to write data to a temporary file,
and delete on exit
:param contents: Data to write. string
:param suffix: File suffix. string
"""
if decompress:
contents = zlib.decompress(contents)
try:
_, fname = tempfile.mkstemp(suffix=suffix)
with open(fname, 'wb') as outfile:
outfile.write(contents)
yield fname
finally:
try:
os.unlink(fname)
except WindowsError: # on Windows the unlink can fail
pass
|
