ENH: Implement_ndim

Julien Marabotto · Julien Marabotto · commit dfb2061bd4d9 · 2024-04-18T15:57:33.000+02:00
Added two tests for ndim: assert nitl.Affine().ndim == 3; assert nitl.LinearTransformsMapping([nitl.Affine()]).ndim == 4 .
diff --git a/nitransforms/tests/test_base.py b/nitransforms/tests/test_base.py
@@ -113,9 +113,8 @@ def _to_hdf5(klass, x5_root):
     )
 
     #Test ndim returned by affine
-    #breakpoint()
     assert nitl.Affine().ndim == 3
-    assert nitl.LinearTransformsMapping(xfm.matrix).ndim == 4
+    assert nitl.LinearTransformsMapping([nitl.Affine()]).ndim == 4
 
     # Test applying to Gifti
     gii = nb.gifti.GiftiImage(

Original file line number	Diff line number	Diff line change
`@@ -113,9 +113,8 @@ def _to_hdf5(klass, x5_root):`
`113`	`113`	`)`
`114`	`114`
`115`	`115`	`#Test ndim returned by affine`
`116`		`- #breakpoint()`
`117`	`116`	`assert nitl.Affine().ndim == 3`
`118`		`- assert nitl.LinearTransformsMapping(xfm.matrix).ndim == 4`
	`117`	`+ assert nitl.LinearTransformsMapping([nitl.Affine()]).ndim == 4`
`119`	`118`
`120`	`119`	`# Test applying to Gifti`
`121`	`120`	`gii = nb.gifti.GiftiImage(`