TEST Test slicing, percentiles

Author: effigies

nibabel/tests/test_spatialimages.py

@@ -24,6 +24,7 @@
 from numpy.testing import assert_array_equal, assert_array_almost_equal
 
 from .test_helpers import bytesio_round_trip
+from .test_viewers import needs_mpl
 from ..testing import (clear_and_catch_warnings, suppress_warnings,
                        VIRAL_MEMMAP)
 from ..tmpdirs import InTemporaryDirectory
@@ -335,6 +336,16 @@ def test_get_data(self):
         assert_false(rt_img.get_data() is out_data)
         assert_array_equal(rt_img.get_data(), in_data)
 
+    @needs_mpl
+    def test_orthoview(self):
+        # Assumes all possible images support int16
+        # See https://github.com/nipy/nibabel/issues/58
+        arr = np.arange(24, dtype=np.int16).reshape((2, 3, 4))
+        img = self.image_class(arr, None)
+        img.orthoview().close()
+        img.orthoview(vlim=(5, 10)).close()
+        img.orthoview(slicer=Ellipsis).close()
+
     def test_api_deprecations(self):
 
         class FakeImage(self.image_class):

nibabel/tests/test_viewers.py

@@ -17,7 +17,7 @@
 from numpy.testing.decorators import skipif
 from numpy.testing import assert_array_equal, assert_equal
 
-from nose.tools import assert_raises, assert_true
+from nose.tools import assert_raises, assert_true, assert_false
 
 # Need at least MPL 1.3 for viewer tests.
 matplotlib, has_mpl, _ = optional_package('matplotlib', min_version='1.3')
@@ -75,6 +75,15 @@ def test_viewer():
     for im in v._ims:
         assert_array_equal(im.get_clim(), vlim)
     assert_array_equal(v._axes[3].get_ylim(), vlim)
+    v.close()
+    v1 = OrthoSlicer3D(data)
+    v2 = OrthoSlicer3D(data, vlim=('1%', '99%'))
+    assert_array_equal(v1.clim, v2.clim)
+    v2.close()
+    v2 = OrthoSlicer3D(data, vlim=('2%', '98%'))
+    assert_false(np.array_equal(v1.clim, v2.clim))
+    v2.close()
+    v1.close()
 
     # non-multi-volume
     v = OrthoSlicer3D(data[:, :, :, 0])
@@ -102,3 +111,15 @@ def test_viewer():
     v2.link_to(v1)  # shouldn't do anything
     v1.close()
     v2.close()
+
+    # Test various slicers
+    OrthoSlicer3D(data, slicer=Ellipsis).close()
+    # Slice time dimension
+    OrthoSlicer3D(data, slicer=(Ellipsis, slice(None, None, 2))).close()
+    OrthoSlicer3D(data, slicer=(Ellipsis, 0)).close()
+    OrthoSlicer3D(data, slicer=(Ellipsis, [0])).close()
+    # Slice spatial dimensions
+    OrthoSlicer3D(data, slicer=(slice(0, 1), slice(0, 1), slice(0, 1))).close()
+    OrthoSlicer3D(data, slicer=(slice(0, -1), slice(0, -1), slice(0, -1))).close()
+    # Fail if we slice too thin
+    assert_raises(ValueError, OrthoSlicer3D, data, slicer=(Ellipsis, 0, 0))