Merge pull request #59 from Eric89GXL/fix-hide-bar

FIX: Fix bar hide in offscreen mode

Author: mwaskom

surfer/tests/test_viz.py

@@ -1,7 +1,7 @@
 import numpy as np
 import os
 from os.path import join as pjoin
-from numpy.testing import assert_raises
+from numpy.testing import assert_raises, assert_array_equal
 from tempfile import mktemp
 import nibabel as nib
 
@@ -34,6 +34,15 @@ def has_freesurfer():
                                     'Requires FreeSurfer command line tools')
 
 
+def test_offscreen():
+    """Test offscreen rendering
+    """
+    mlab.options.backend = 'auto'
+    brain = Brain(*std_args, offscreen=True)
+    shot = brain.screenshot()
+    assert_array_equal(shot.shape, (800, 800, 3))
+
+
 def test_image():
     """Test image saving
     """

surfer/viz.py

@@ -562,13 +562,13 @@ def toggle_toolbars(self, show=None):
                 if hasattr(bars[0], 'isVisible'):
                     # QT4
                     show = not bars[0].isVisible()
-                else:
+                elif hasattr(bars[0], 'Shown'):
                     # WX
                     show = not bars[0].Shown()
             for bar in bars:
                 if hasattr(bar, 'setVisible'):
                     bar.setVisible(show)
-                else:
+                elif hasattr(bar, 'Show'):
                     bar.Show(show)
 
     def _get_one_brain(self, d, name):