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
|
# Authors: Eric Larson <larson.eric.d@gmail.com>
#
# License: Simplified BSD
import os.path as op
import warnings
from numpy.testing import assert_raises
from mne import io, read_events, pick_types
from mne.datasets import sample
warnings.simplefilter('always') # enable b/c these tests throw warnings
# Set our plotters to test mode
import matplotlib
matplotlib.use('Agg') # for testing don't use X server
import matplotlib.pyplot as plt
data_dir = sample.data_path(download=False)
subjects_dir = op.join(data_dir, 'subjects')
base_dir = op.join(op.dirname(__file__), '..', '..', 'io', 'tests', 'data')
raw_fname = op.join(base_dir, 'test_raw.fif')
event_name = op.join(base_dir, 'test-eve.fif')
def _fake_click(fig, ax, point, xform='ax'):
"""Helper to fake a click at a relative point within axes"""
if xform == 'ax':
x, y = ax.transAxes.transform_point(point)
elif xform == 'data':
x, y = ax.transData.transform_point(point)
else:
raise ValueError('unknown transform')
try:
fig.canvas.button_press_event(x, y, 1, False, None)
except: # for old MPL
fig.canvas.button_press_event(x, y, 1, False)
def _get_raw():
raw = io.Raw(raw_fname, preload=True)
raw.pick_channels(raw.ch_names[:9])
return raw
def _get_events():
return read_events(event_name)
def test_plot_raw():
"""Test plotting of raw data
"""
raw = _get_raw()
events = _get_events()
plt.close('all') # ensure all are closed
with warnings.catch_warnings(record=True):
fig = raw.plot(events=events, show_options=True)
# test mouse clicks
x = fig.get_axes()[0].lines[1].get_xdata().mean()
y = fig.get_axes()[0].lines[1].get_ydata().mean()
data_ax = fig.get_axes()[0]
_fake_click(fig, data_ax, [x, y], xform='data') # mark a bad channel
_fake_click(fig, data_ax, [x, y], xform='data') # unmark a bad channel
_fake_click(fig, data_ax, [0.5, 0.999]) # click elsewhere in 1st axes
_fake_click(fig, fig.get_axes()[1], [0.5, 0.5]) # change time
_fake_click(fig, fig.get_axes()[2], [0.5, 0.5]) # change channels
_fake_click(fig, fig.get_axes()[3], [0.5, 0.5]) # open SSP window
fig.canvas.button_press_event(1, 1, 1) # outside any axes
# sadly these fail when no renderer is used (i.e., when using Agg):
#ssp_fig = set(plt.get_fignums()) - set([fig.number])
#assert_equal(len(ssp_fig), 1)
#ssp_fig = plt.figure(list(ssp_fig)[0])
#ax = ssp_fig.get_axes()[0] # only one axis is used
#t = [c for c in ax.get_children() if isinstance(c,
# matplotlib.text.Text)]
#pos = np.array(t[0].get_position()) + 0.01
#_fake_click(ssp_fig, ssp_fig.get_axes()[0], pos, xform='data') # off
#_fake_click(ssp_fig, ssp_fig.get_axes()[0], pos, xform='data') # on
# test keypresses
fig.canvas.key_press_event('escape')
fig.canvas.key_press_event('down')
fig.canvas.key_press_event('up')
fig.canvas.key_press_event('right')
fig.canvas.key_press_event('left')
fig.canvas.key_press_event('o')
fig.canvas.key_press_event('escape')
plt.close('all')
def test_plot_raw_psds():
"""Test plotting of raw psds
"""
import matplotlib.pyplot as plt
raw = _get_raw()
# normal mode
raw.plot_psds(tmax=2.0)
# specific mode
picks = pick_types(raw.info, meg='mag', eeg=False)[:4]
raw.plot_psds(picks=picks, area_mode='range')
ax = plt.axes()
# if ax is supplied, picks must be, too:
assert_raises(ValueError, raw.plot_psds, ax=ax)
raw.plot_psds(picks=picks, ax=ax)
plt.close('all')
|
