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
89
90
91
92
93
|
import os.path as op
import sys
from subprocess import Popen, PIPE
from mne.utils import run_tests_if_main
run_script = """
from __future__ import print_function
import sys
import mne
out = []
# check scipy
ok_scipy_submodules = set(['scipy', 'numpy', # these appear in old scipy
'fftpack', 'lib', 'linalg',
'misc', 'sparse', 'version'])
scipy_submodules = set(x.split('.')[1] for x in sys.modules.keys()
if x.startswith('scipy.') and '__' not in x and
not x.split('.')[1].startswith('_')
and sys.modules[x] is not None)
bad = scipy_submodules - ok_scipy_submodules
if len(bad) > 0:
out.append('scipy submodules: %s' % list(bad))
# check sklearn and others
_sklearn = _pandas = _mayavi = _matplotlib = False
for x in sys.modules.keys():
if x.startswith('sklearn') and not _sklearn:
out.append('sklearn')
_sklearn = True
if x.startswith('pandas') and not _pandas:
out.append('pandas')
_pandas = True
if x.startswith('mayavi') and not _mayavi:
out.append('mayavi')
_mayavi = True
if x.startswith('matplotlib') and not _matplotlib:
out.append('matplotlib')
_matplotlib = True
if len(out) > 0:
print('\\nFound un-nested imports for:\\n' + '\\n'.join(out), end='')
exit(1)
"""
def test_module_nesting():
"""Test that module imports are properly nested."""
proc = Popen([sys.executable, '-c', run_script], stdout=PIPE, stderr=PIPE)
stdout, stderr = proc.communicate()
stdout = stdout.decode('utf-8')
stderr = stderr.decode('utf-8')
assert not proc.returncode, stdout + stderr
mpl_script = """
import os
import os.path as op
import re
import sys
import mne
reg = re.compile('test_.*.py')
for dirpath, _, filenames in os.walk('{0}'):
if dirpath.endswith('tests'):
test_dir = op.join('{0}', dirpath)
sys.path.insert(0, test_dir)
for filename in filenames:
if reg.match(filename) is not None:
__import__(op.splitext(filename)[0])
for x in sys.modules.keys():
if x.startswith('matplotlib.pyplot'):
print('\\nFound un-nested pyplot import: ' +
op.join(test_dir, filename))
exit(1)
sys.path.pop(0)
"""
def test_mpl_nesting():
"""Test that matplotlib imports are properly nested in tests."""
mne_path = op.abspath(op.join(op.dirname(__file__), '..'))
proc = Popen([sys.executable, '-c', mpl_script.format(mne_path)],
stdout=PIPE, stderr=PIPE)
stdout, stderr = proc.communicate()
stdout = stdout.decode('utf-8')
stderr = stderr.decode('utf-8')
assert not proc.returncode, stdout + stderr
run_tests_if_main()
|
