|
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