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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
from contextlib import nullcontext
import os
import subprocess
import sys
import pytest
import mne
from mne.utils import sizeof_fmt, run_subprocess, catch_logging
def test_sizeof_fmt():
"""Test sizeof_fmt."""
assert sizeof_fmt(0) == '0 bytes'
assert sizeof_fmt(1) == '1 byte'
assert sizeof_fmt(1000) == '1000 bytes'
def test_html_repr():
"""Test switching _repr_html_ between HTML and plain text."""
key = "MNE_REPR_HTML"
existing_value = os.getenv(key, None)
os.environ[key] = "True" # HTML repr on
info = mne.create_info(10, 256)
r = info._repr_html_()
assert r.startswith("<table")
assert r.endswith("</table>")
os.environ[key] = "False" # HTML repr off
r = info._repr_html_()
assert r.startswith("<pre>")
assert r.endswith("</pre>")
del os.environ[key]
if existing_value is not None:
os.environ[key, existing_value]
@pytest.mark.parametrize('kind', ('stdout', 'stderr'))
@pytest.mark.parametrize('do_raise', (True, False))
def test_run_subprocess(tmp_path, capsys, kind, do_raise):
"""Test run_subprocess."""
fname = tmp_path / 'subp.py'
extra = ''
if do_raise:
extra = """
raise RuntimeError('This is a test')
"""
raise_context = pytest.raises(subprocess.CalledProcessError)
else:
extra = ''
raise_context = nullcontext()
with open(fname, 'w') as fid:
fid.write(f"""\
import sys
import time
print('foo', file=sys.{kind})
print('bar', file=sys.{kind})
""" + extra)
with catch_logging() as log, raise_context:
stdout, stderr = run_subprocess(
[sys.executable, str(fname)], verbose=True)
if do_raise:
exc = raise_context.excinfo.value
stdout = exc.stdout
stderr = exc.stderr
log = log.getvalue()
log = '\n'.join(log.split('\n')[1:]) # get rid of header
log = log.replace('\r\n', '\n') # Windows
orig_log = log
stdout = stdout.replace('\r\n', '\n')
stderr = stderr.replace('\r\n', '\n')
if do_raise: # remove traceback
def remove_traceback(log):
return '\n'.join(
line for line in log.split('\n')
if not line.strip().startswith(
('File ', 'raise ', 'RuntimeError: ', 'Traceback ')))
log = remove_traceback(log)
stderr = remove_traceback(stderr)
want = 'foo\nbar\n'
assert log == want, orig_log
if kind == 'stdout':
std = stdout
other = stderr
else:
std = stderr
other = stdout
assert std == want
assert other == ''
stdout, stderr = capsys.readouterr()
assert stdout == ''
assert stderr == ''
# Now make sure we can pass other stuff as stdout/stderr
capsys.readouterr() # clear
stdout_fname = tmp_path / 'stdout.txt'
stderr_fname = tmp_path / 'stderr.txt'
stdout_file = open(stdout_fname, 'w')
stderr_file = open(stderr_fname, 'w')
if do_raise:
raise_context = pytest.raises(subprocess.CalledProcessError)
else:
raise_context = nullcontext()
with catch_logging() as log, stdout_file, stderr_file, raise_context:
stdout, stderr = run_subprocess(
[sys.executable, str(fname)], verbose=False,
stdout=stdout_file, stderr=stderr_file)
if do_raise:
exc = raise_context.excinfo.value
assert exc.stdout is None
assert exc.stderr is None
else:
assert stdout == ''
assert stderr == ''
log = log.getvalue()
assert log == ''
stdout, stderr = capsys.readouterr()
assert stdout == ''
assert stderr == ''
stdout = stdout_fname.read_text()
stderr = stderr_fname.read_text()
if do_raise:
stderr = '\n'.join(stderr.split('\n')[:-5])
if stderr:
stderr += '\n'
if kind == 'stdout':
std = stdout
other = stderr
else:
std = stderr
other = stdout
assert std == want
assert other == ''
|
