enh: add a test on ndim

Author: Julien Marabotto

nitransforms/base.py

@@ -15,7 +15,6 @@
 from nibabel import funcs as _nbfuncs
 from nibabel.nifti1 import intent_codes as INTENT_CODES
 from nibabel.cifti2 import Cifti2Image
-from nitransforms import linear as nitl
 
 EQUALITY_TOL = 1e-5
 

nitransforms/tests/test_base.py

@@ -114,7 +114,9 @@ def _to_hdf5(klass, x5_root):
 
     #Test ndim returned by affine
     assert nitl.Affine().ndim == 3
-    assert nitl.LinearTransformsMapping([nitl.Affine()]).ndim == 4
+    assert nitl.LinearTransformsMapping(
+        [nitl.Affine(), nitl.Affine()]
+    ).ndim == 4
 
     # Test applying to Gifti
     gii = nb.gifti.GiftiImage(