TEST: Add test for MakeRibbon interface

Author: effigies

smriprep/interfaces/tests/__init__.py

@@ -0,0 +1,40 @@
+import nibabel as nb
+from nipype.pipeline import engine as pe
+import numpy as np
+
+from ...data import load_resource
+from ..surf import MakeRibbon
+
+
+def test_MakeRibbon(tmp_path):
+    res_template = "{path}/sub-fsaverage_res-4_hemi-{hemi}_desc-cropped_{surf}dist.nii.gz"
+    white, pial = [
+        [
+            load_resource(
+                res_template.format(path="../interfaces/tests/data", hemi=hemi, surf=surf)
+            )
+            for hemi in "LR"
+        ]
+        for surf in ("wm", "pial")
+    ]
+
+    make_ribbon = pe.Node(
+        MakeRibbon(white_distvols=white, pial_distvols=pial),
+        name="make_ribbon",
+        base_dir=tmp_path,
+    )
+
+    result = make_ribbon.run()
+
+    ribbon = nb.load(result.outputs.ribbon)
+    expected = nb.load(
+        load_resource(
+            load_resource(
+                "../interfaces/tests/data/sub-fsaverage_res-4_desc-cropped_ribbon.nii.gz"
+            )
+        )
+    )
+
+    assert ribbon.shape == expected.shape
+    assert np.allclose(ribbon.affine, expected.affine)
+    assert np.array_equal(ribbon.dataobj, expected.dataobj)