32
32
def setup_data_env ():
33
33
setup_environment ()
34
34
global DATA_FUNCS
35
+ DATA_FUNCS ['home_dir_func' ] = nibd .get_nipy_user_dir
35
36
DATA_FUNCS ['sys_dir_func' ] = nibd .get_nipy_system_dir
36
37
DATA_FUNCS ['path_func' ] = nibd .get_data_path
37
38
38
39
39
40
def teardown_data_env ():
40
41
teardown_environment ()
42
+ nibd .get_nipy_user_dir = DATA_FUNCS ['home_dir_func' ]
41
43
nibd .get_nipy_system_dir = DATA_FUNCS ['sys_dir_func' ]
42
44
nibd .get_data_path = DATA_FUNCS ['path_func' ]
43
45
@@ -135,7 +137,9 @@ def test_data_path():
135
137
del env [DATA_KEY ]
136
138
if USER_KEY in env :
137
139
del os .environ [USER_KEY ]
140
+ fake_user_dir = '/user/path'
138
141
nibd .get_nipy_system_dir = lambda : ''
142
+ nibd .get_nipy_user_dir = lambda : fake_user_dir
139
143
# now we should only have anything pointed to in the user's dir
140
144
old_pth = get_data_path ()
141
145
# We should have only sys.prefix and, iff sys.prefix == /usr,
@@ -144,8 +148,7 @@ def test_data_path():
144
148
def_dirs = [pjoin (sys .prefix , 'share' , 'nipy' )]
145
149
if sys .prefix == '/usr' :
146
150
def_dirs .append (pjoin ('/usr/local' , 'share' , 'nipy' ))
147
- home_nipy = pjoin (os .path .expanduser ('~' ), '.nipy' )
148
- assert_equal (old_pth , def_dirs + [home_nipy ])
151
+ assert_equal (old_pth , def_dirs + ['/user/path' ])
149
152
# then we'll try adding some of our own
150
153
tst_pth = '/a/path' + os .path .pathsep + '/b/ path'
151
154
tst_list = ['/a/path' , '/b/ path' ]
@@ -161,9 +164,9 @@ def test_data_path():
161
164
with open (tmpfile , 'wt' ) as fobj :
162
165
fobj .write ('[DATA]\n ' )
163
166
fobj .write ('path = %s' % tst_pth )
164
- os . environ [ USER_KEY ] = tmpdir
167
+ nibd . get_nipy_user_dir = lambda : tmpdir
165
168
assert_equal (get_data_path (), tst_list + def_dirs + [tmpdir ])
166
- del os . environ [ USER_KEY ]
169
+ nibd . get_nipy_user_dir = lambda : fake_user_dir
167
170
assert_equal (get_data_path (), old_pth )
168
171
# with some trepidation, the system config files
169
172
with TemporaryDirectory () as tmpdir :
0 commit comments