1
1
import numpy as np
2
2
import os
3
+ import os .path as op
3
4
from os .path import join as pjoin
4
5
from numpy .testing import assert_raises , assert_array_equal
5
6
from tempfile import mktemp
12
13
subj_dir = utils ._get_subjects_dir ()
13
14
subject_id = 'fsaverage'
14
15
std_args = [subject_id , 'lh' , 'inflated' ]
15
- data_dir = pjoin (os .path .split (__file__ )[0 ], '..' , '..' ,
16
- 'examples' , 'example_data' )
16
+ data_dir = pjoin (op .dirname (__file__ ), '..' , '..' , 'examples' , 'example_data' )
17
17
small_brain = dict (size = 100 )
18
18
19
+ overlay_fname = pjoin (data_dir , 'lh.sig.nii.gz' )
19
20
20
21
def has_freesurfer ():
21
22
if 'FREESURFER_HOME' not in os .environ :
@@ -42,11 +43,16 @@ def test_offscreen():
42
43
def test_image ():
43
44
"""Test image saving
44
45
"""
46
+ tmp_name = mktemp () + '.png'
47
+
45
48
mlab .options .backend = 'auto'
49
+ subject_id , _ , surf = std_args
50
+ brain = Brain (subject_id , 'both' , surf = surf , config_opts = small_brain )
51
+ brain .add_overlay (overlay_fname , hemi = 'lh' , min = 5 , max = 20 , sign = "pos" )
52
+ brain .save_imageset (tmp_name , ['med' , 'lat' ], 'jpg' )
53
+
46
54
brain = Brain (* std_args , config_opts = small_brain )
47
- tmp_name = mktemp () + '.png'
48
55
brain .save_image (tmp_name )
49
- brain .save_imageset (tmp_name , ['med' , 'lat' ], 'jpg' )
50
56
brain .save_montage (tmp_name , ['l' , 'v' , 'm' ], orientation = 'v' )
51
57
brain .save_montage (tmp_name , ['l' , 'v' , 'm' ], orientation = 'h' )
52
58
brain .save_montage (tmp_name , [['l' , 'v' ], ['m' , 'f' ]])
0 commit comments