FIX: Fix orientations and interactions

Author: larsoner

nibabel/tests/test_viewers.py

@@ -14,6 +14,7 @@
 from ..viewers import OrthoSlicer3D
 
 from numpy.testing.decorators import skipif
+from numpy.testing import assert_array_equal
 
 from nose.tools import assert_raises
 
@@ -29,7 +30,7 @@ def test_viewer():
     data = (np.outer(a, b)[..., np.newaxis] * a)[:, :, :, np.newaxis]
     data = data * np.array([1., 2.])  # give it a # of volumes > 1
     v = OrthoSlicer3D(data)
-    plt.draw()
+    assert_array_equal(v.position, (0, 0, 0))
 
     # fake some events, inside and outside axes
     v._on_scroll(nt('event', 'button inaxes key')('up', None, None))
@@ -41,6 +42,8 @@ def test_viewer():
     for ax in v._axes.values():
         v._on_mouse(nt('event', 'xdata ydata inaxes button')(0.5, 0.5, ax, 1))
     v._on_mouse(nt('event', 'xdata ydata inaxes button')(0.5, 0.5, None, None))
+    v.set_volume_idx(1)
+    v.set_volume_idx(1)  # should just pass
     v.close()
 
     # non-multi-volume
@@ -51,7 +54,14 @@ def test_viewer():
     # other cases
     fig, axes = plt.subplots(1, 4)
     plt.close(fig)
-    OrthoSlicer3D(data, pcnt_range=[0.1, 0.9], axes=axes)
-    OrthoSlicer3D(data, axes=axes[:3])
+    v1 = OrthoSlicer3D(data, pcnt_range=[0.1, 0.9], axes=axes)
+    aff = np.array([[0, 1, 0, 3], [-1, 0, 0, 2], [0, 0, 2, 1], [0, 0, 0, 1]],
+                   float)
+    v2 = OrthoSlicer3D(data, affine=aff, axes=axes[:3])
     assert_raises(ValueError, OrthoSlicer3D, data[:, :, 0, 0])
     assert_raises(ValueError, OrthoSlicer3D, data, affine=np.eye(3))
+    assert_raises(TypeError, v2.link_to, 1)
+    v2.link_to(v1)
+    v2.link_to(v1)  # shouldn't do anything
+    v1.close()
+    v2.close()