Skip to content

Commit d0f71e5

Browse files
Update testing
1 parent adf37f6 commit d0f71e5

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

mne/viz/_brain/tests/test_notebook.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ def test_notebook_alignment(renderer_notebook, brain_gc, nbexec):
3838
def test_notebook_interactive(renderer_notebook, brain_gc, nbexec):
3939
"""Test interactive modes."""
4040
import os
41+
import tempfile
4142
from contextlib import contextmanager
4243
from numpy.testing import assert_allclose
4344
from ipywidgets import Button
@@ -72,6 +73,11 @@ def interactive(on):
7273
assert brain._renderer.figure.notebook
7374
assert brain._renderer.figure.display is not None
7475
brain._renderer._update()
76+
tmp_path = tempfile.mkdtemp()
77+
movie_path = os.path.join(tmp_path, 'test.gif')
78+
screenshot_path = os.path.join(tmp_path, 'test.png')
79+
brain._renderer.actions['movie_field'].value = movie_path
80+
brain._renderer.actions['screenshot_field'].value = screenshot_path
7581
total_number_of_buttons = sum(
7682
'_field' not in k for k in brain._renderer.actions.keys())
7783
number_of_buttons = 0
@@ -80,6 +86,8 @@ def interactive(on):
8086
action.click()
8187
number_of_buttons += 1
8288
assert number_of_buttons == total_number_of_buttons
89+
assert os.path.isfile(movie_path)
90+
assert os.path.isfile(screenshot_path)
8391
img_nv = brain.screenshot()
8492
assert img_nv.shape == (300, 300, 3), img_nv.shape
8593
img_v = brain.screenshot(time_viewer=True)

0 commit comments

Comments
 (0)