TEST: Check dim_info reorientation

Author: effigies

nibabel/tests/test_nifti1.py

@@ -28,6 +28,7 @@
 from nibabel.spatialimages import HeaderDataError
 from nibabel.tmpdirs import InTemporaryDirectory
 from ..freesurfer import load as mghload
+from ..orientations import axcodes2ornt
 
 from .test_arraywriters import rt_err_estimate, IUINT_TYPES
 from .test_helpers import bytesio_filemap, bytesio_round_trip
@@ -1403,6 +1404,38 @@ def test_rt_bias(self):
                 bias_thresh = np.max([max_miss / np.sqrt(count), eps])
                 assert_true(np.abs(bias) < bias_thresh)
 
+    def test_reoriented_dim_info(self):
+        # Check that dim_info is reoriented correctly
+        arr = np.arange(24).reshape((2, 3, 4))
+        # Start as RAS
+        aff = np.diag([2, 3, 4, 1])
+        simg = self.single_class(arr, aff)
+        axcodes = ('RAS', 'RPS',  # Include a flip
+                   'RSA', 'RSP',  # Include a flip + reorientation
+                   'ARS',
+                   'ASR',
+                   'SRA',
+                   'SAR')
+        for info in ((0, 1, 2),
+                     (0, 2, 1),
+                     (None, None, None),
+                     (0, 2, None),
+                     (0, None, None),
+                     (None, 2, 1),
+                     (None, None, 1),
+                     ):
+            simg.header.set_dim_info(*info)
+            for axcode in axcodes:
+                ornt = axcodes2ornt(axcode)
+                reorder = ornt[:, 0].astype(int)
+                rimg = simg.as_reoriented(ornt)
+                # Check shape and zooms to ensure the test is correct
+                assert_equal(simg.shape, tuple(np.array(rimg.shape)[reorder]))
+                assert_equal(simg.header.get_zooms(),
+                             tuple(np.array(rimg.header.get_zooms())[reorder]))
+                assert_equal(simg.header.get_dim_info(),
+                             tuple(np.array(rimg.header.get_dim_info())[reorder]))
+
 
 @runif_extra_has('slow')
 def test_large_nifti1():