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
|
# Authors: Eric Larson <larson.eric.d@gmail.com>
#
# License: BSD (3-clause)
import numpy as np
from nose.tools import assert_equal, assert_raises
from mne import create_info
from mne.io import RawArray
from mne.utils import logger, catch_logging, slow_test, run_tests_if_main
def bad_1(x):
return # bad return type
def bad_2(x):
return x[:-1] # bad shape
def printer(x):
logger.info('exec')
return x
@slow_test
def test_apply_function_verbose():
"""Test apply function verbosity
"""
n_chan = 2
n_times = 3
ch_names = [str(ii) for ii in range(n_chan)]
raw = RawArray(np.zeros((n_chan, n_times)),
create_info(ch_names, 1., 'mag'))
# test return types in both code paths (parallel / 1 job)
assert_raises(TypeError, raw.apply_function, bad_1,
None, None, 1)
assert_raises(ValueError, raw.apply_function, bad_2,
None, None, 1)
assert_raises(TypeError, raw.apply_function, bad_1,
None, None, 2)
assert_raises(ValueError, raw.apply_function, bad_2,
None, None, 2)
# check our arguments
with catch_logging() as sio:
raw.apply_function(printer, None, None, 1, verbose=False)
assert_equal(len(sio.getvalue()), 0)
raw.apply_function(printer, None, None, 1, verbose=True)
assert_equal(sio.getvalue().count('\n'), n_chan)
run_tests_if_main()
|
