|
13 | 13 | from unittest import SkipTest |
14 | 14 |
|
15 | 15 | from surfer import Brain, io, utils |
16 | | -from surfer.utils import requires_fsaverage, requires_imageio, requires_fs |
| 16 | +from surfer.utils import (requires_fsaverage, requires_imageio, requires_fs, |
| 17 | + _get_extra) |
17 | 18 |
|
18 | 19 | warnings.simplefilter('always') |
19 | 20 |
|
@@ -247,30 +248,30 @@ def test_label(): |
247 | 248 | brain = Brain(subject_id, hemi, surf) |
248 | 249 | view = get_view(brain) |
249 | 250 |
|
250 | | - brain.add_label("BA1") |
| 251 | + extra, subj_dir = _get_extra() |
| 252 | + brain.add_label("BA1" + extra) |
251 | 253 | check_view(brain, view) |
252 | | - brain.add_label("BA1", color="blue", scalar_thresh=.5) |
253 | | - subj_dir = utils._get_subjects_dir() |
| 254 | + brain.add_label("BA1" + extra, color="blue", scalar_thresh=.5) |
254 | 255 | label_file = pjoin(subj_dir, subject_id, |
255 | | - "label", "%s.MT.label" % hemi) |
| 256 | + "label", "%s.MT%s.label" % (hemi, extra)) |
256 | 257 | brain.add_label(label_file) |
257 | | - brain.add_label("BA44", borders=True) |
258 | | - brain.add_label("BA6", alpha=.7) |
| 258 | + brain.add_label("BA44" + extra, borders=True) |
| 259 | + brain.add_label("BA6" + extra, alpha=.7) |
259 | 260 | brain.show_view("medial") |
260 | | - brain.add_label("V1", color="steelblue", alpha=.6) |
261 | | - brain.add_label("V2", color="#FF6347", alpha=.6) |
262 | | - brain.add_label("entorhinal", color=(.2, 1, .5), alpha=.6) |
| 261 | + brain.add_label("V1" + extra, color="steelblue", alpha=.6) |
| 262 | + brain.add_label("V2" + extra, color="#FF6347", alpha=.6) |
| 263 | + brain.add_label("entorhinal" + extra, color=(.2, 1, .5), alpha=.6) |
263 | 264 | brain.set_surf('white') |
264 | 265 | brain.show_view(dict(elevation=40, distance=430), distance=430) |
265 | 266 | with pytest.raises(ValueError, match='!='): |
266 | 267 | brain.show_view(dict(elevation=40, distance=430), distance=431) |
267 | 268 |
|
268 | 269 | # remove labels |
269 | | - brain.remove_labels('V1') |
270 | | - assert 'V2' in brain.labels_dict |
271 | | - assert 'V1' not in brain.labels_dict |
| 270 | + brain.remove_labels('V1' + extra) |
| 271 | + assert 'V2' + extra in brain.labels_dict |
| 272 | + assert 'V1' + extra not in brain.labels_dict |
272 | 273 | brain.remove_labels() |
273 | | - assert 'V2' not in brain.labels_dict |
| 274 | + assert 'V2' + extra not in brain.labels_dict |
274 | 275 |
|
275 | 276 | brain.close() |
276 | 277 |
|
@@ -354,7 +355,8 @@ def test_morphometry(): |
354 | 355 | def test_movie(tmpdir): |
355 | 356 | """Test saving a movie of an MEG inverse solution.""" |
356 | 357 | import imageio |
357 | | - |
| 358 | + if sys.version_info < (3,): |
| 359 | + raise SkipTest('imageio ffmpeg requires Python 3') |
358 | 360 | # create and setup the Brain instance |
359 | 361 | _set_backend() |
360 | 362 | brain = Brain(*std_args) |
@@ -432,15 +434,15 @@ def test_probabilistic_labels(): |
432 | 434 | brain = Brain("fsaverage", "lh", "inflated", |
433 | 435 | cortex="low_contrast") |
434 | 436 |
|
435 | | - brain.add_label("BA1", color="darkblue") |
436 | | - |
437 | | - brain.add_label("BA1", color="dodgerblue", scalar_thresh=.5) |
| 437 | + extra, subj_dir = _get_extra() |
| 438 | + brain.add_label("BA1" + extra, color="darkblue") |
| 439 | + brain.add_label("BA1" + extra, color="dodgerblue", scalar_thresh=.5) |
| 440 | + brain.add_label("BA45" + extra, color="firebrick", borders=True) |
| 441 | + brain.add_label("BA45" + extra, color="salmon", borders=True, |
| 442 | + scalar_thresh=.5) |
438 | 443 |
|
439 | | - brain.add_label("BA45", color="firebrick", borders=True) |
440 | | - brain.add_label("BA45", color="salmon", borders=True, scalar_thresh=.5) |
441 | | - |
442 | | - subj_dir = utils._get_subjects_dir() |
443 | | - label_file = pjoin(subj_dir, "fsaverage", "label", "lh.BA6.label") |
| 444 | + label_file = pjoin(subj_dir, "fsaverage", "label", |
| 445 | + "lh.BA6%s.label" % (extra,)) |
444 | 446 | prob_field = np.zeros_like(brain.geo['lh'].x) |
445 | 447 | ids, probs = nib.freesurfer.read_label(label_file, read_scalars=True) |
446 | 448 | prob_field[ids] = probs |
|
0 commit comments