Skip to content

Commit 5de504a

Browse files
committed
fix test for Travis and Circle
1 parent 4dc5896 commit 5de504a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

nipype/utils/tests/test_config.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,15 @@ def test_display_parse(monkeypatch, disp_var):
3232
"""Check that when $DISPLAY is defined, the display is correctly parsed"""
3333
config._display = None
3434
dispstr = ':0'
35-
monkeypatch.setenv('DISPLAY', disp_var)
35+
old_disp = os.getenv('DISPLAY')
36+
os.environ['DISPLAY'] = disp_var
3637
assert config.get_display() == dispstr
3738
# Test that it was correctly cached
3839
assert config.get_display() == dispstr
40+
if old_disp is not None:
41+
os.environ['DISPLAY'] = old_disp
42+
else:
43+
del os.environ['DISPLAY']
3944

4045

4146
@pytest.mark.parametrize('dispnum', range(5))

0 commit comments

Comments
 (0)