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
137
138
139
140
141
142
143
144
|
# -*- coding: utf-8 -*-
# Authors: Alex Rockhill <aprockhill@mailbox.org>
#
# License: Simplified BSD
from pathlib import Path
import sys
import pytest
from mne.viz.backends.renderer import _get_backend
from mne.viz.backends.tests._utils import skips_if_not_pyvistaqt
from mne.utils import check_version
def _do_widget_tests(backend):
# testing utils
widget_checks = set()
def callback(x=None):
add = x
if add is None:
add = 'click'
elif isinstance(add, str):
add = add.lstrip('&') # new notebooks can add this
widget_checks.add(add)
window = backend._AppWindow()
central_layout = backend._VBoxLayout(scroll=(500, 500))
renderer = backend._3DRenderer(name='test', size=(200, 200))
renderer.sphere([0, 0, 0], 'red', 1)
central_layout._add_widget(renderer.plotter)
canvas = backend._Canvas(5, 5, 96)
canvas.ax.plot(range(10), range(10), label='plot')
central_layout._add_widget(canvas)
central_layout._add_widget(backend._Label('test'))
text = backend._Text('test', 'placeholder', callback)
central_layout._add_widget(text)
button = backend._Button('test2', callback)
central_layout._add_widget(button)
slider = backend._Slider(0, (0, 100), callback)
central_layout._add_widget(slider)
checkbox = backend._CheckBox(0, callback)
central_layout._add_widget(checkbox)
spinbox = backend._SpinBox(10, (5, 50), callback, step=4)
central_layout._add_widget(spinbox)
combobox = backend._ComboBox('40', ('5', '50', '40'), callback)
central_layout._add_widget(combobox)
radio_buttons = backend._RadioButtons('40', ('5', '50', '40'), callback)
central_layout._add_widget(radio_buttons)
groupbox = backend._GroupBox(
'menu', [backend._Button(f'{i}', callback) for i in range(5)])
central_layout._add_widget(groupbox)
groupbox._set_enabled(False)
groupbox._set_enabled(True)
file_button = backend._FileButton(callback, window=window)
central_layout._add_widget(file_button)
play_menu = backend._PlayMenu(0, (0, 100), callback)
central_layout._add_widget(play_menu)
progress_bar = backend._ProgressBar(100)
progress_bar._increment()
central_layout._add_widget(progress_bar)
window._add_keypress(callback)
window._set_central_layout(central_layout)
window._set_focus()
window._show()
# pop up
popup = backend._Popup('Info', 'this is a message', 'test', callback)
# do tests
# first, test popup
popup._click('Ok')
assert 'Ok' in widget_checks
window._trigger_keypress('a')
assert 'a' in widget_checks
window._trigger_keypress('escape')
assert 'escape' in widget_checks
# test each widget
text._set_value('foo')
assert 'foo' in widget_checks
button._click()
assert 'click' in widget_checks
slider._set_value(10)
assert 10 in widget_checks
checkbox._set_checked(True)
assert True in widget_checks
spinbox._set_value(20)
assert 20 in widget_checks
combobox._set_value('5')
assert '5' in widget_checks
radio_buttons._set_value('50')
assert '50' in widget_checks
# this was tested manually but creates a blocking window so can't
# be tested here
# file_button.click()
assert hasattr(file_button, 'click')
play_menu._set_value(99)
assert 99 in widget_checks
window._close()
@skips_if_not_pyvistaqt
def test_widget_abstraction_pyvistaqt(renderer_pyvistaqt):
"""Test the GUI widgets abstraction."""
backend = _get_backend()
assert Path(backend.__file__).stem == '_qt'
_do_widget_tests(backend)
nb_skip_mark = pytest.mark.skipif(
sys.platform.startswith('win') or not check_version('ipympl'),
reason='need ipympl and nbexec does not work on Windows')
# Marking directly with skipif causes problems for nbexec, so let's get it in
# via a param
@pytest.mark.parametrize('skippy', [pytest.param('', marks=nb_skip_mark)])
def test_widget_abstraction_notebook(renderer_notebook, nbexec, skippy):
"""Test the GUI widgets abstraction in notebook."""
from pathlib import Path
from mne.viz import set_3d_backend
from mne.viz.backends.renderer import _get_backend
from mne.viz.backends.tests.test_abstract import _do_widget_tests
from IPython import get_ipython
set_3d_backend('notebook')
backend = _get_backend()
assert Path(backend.__file__).stem == '_notebook'
ipython = get_ipython()
ipython.magic('%matplotlib widget')
_do_widget_tests(backend)
|
