BF: fixed test error when DISPLAY variable not available

satra · satra · commit 35a143981a97 · 2011-01-18T18:11:55.000-05:00
diff --git a/nipype/interfaces/freesurfer/tests/test_utils.py b/nipype/interfaces/freesurfer/tests/test_utils.py
@@ -227,10 +227,13 @@ def test_surfshots():
     yield assert_equal, fotos.cmdline, "tksurfer fsaverage lh pial -tcl other.tcl"
 
     # Test that the interface crashes politely if graphics aren't enabled
-    hold_display = os.environ["DISPLAY"]
-    del os.environ["DISPLAY"]
-    yield assert_raises, RuntimeError, fotos.run
-    os.environ["DISPLAY"] = hold_display
-
+    try:
+        hold_display = os.environ["DISPLAY"]
+        del os.environ["DISPLAY"]
+        yield assert_raises, RuntimeError, fotos.run
+        os.environ["DISPLAY"] = hold_display
+    except KeyError:
+        pass
+    
     # Clean up our mess
     clean_directory(cwd, oldwd)
