FIX: raise TypeError on complex input prior to figure creation

Author: grlee77

nibabel/tests/test_viewers.py

@@ -66,6 +66,11 @@ def test_viewer():
     v = OrthoSlicer3D(data[:, :, :, 0])
     v._on_scroll(nt('event', 'button inaxes key')('up', v._axes[0], 'shift'))
     v._on_keypress(nt('event', 'key')('escape'))
+    v.close()
+
+    # complex input should raise a TypeError prior to figure creation
+    assert_raises(TypeError, OrthoSlicer3D,
+                  data[:, :, :, 0].astype(np.complex64))
 
     # other cases
     fig, axes = plt.subplots(1, 4)

nibabel/viewers.py

@@ -72,6 +72,8 @@ def __init__(self, data, affine=None, axes=None, cmap='gray',
         data = np.asanyarray(data)
         if data.ndim < 3:
             raise ValueError('data must have at least 3 dimensions')
+        if np.iscomplexobj(data):
+            raise TypeError("Complex data not supported")
         affine = np.array(affine, float) if affine is not None else np.eye(4)
         if affine.ndim != 2 or affine.shape != (4, 4):
             raise ValueError('affine must be a 4x4 matrix')