[MAINT] Restart circle

dPys · Sep 25, 2020 · cf47579 · cf47579
1 parent 7635dd2
commit cf47579
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tests/test_clustering.py b/tests/test_clustering.py
@@ -67,7 +67,7 @@ def test_make_local_connectivity_tcorr():
 
     out_img = clustools.parcellate_ncut(W, 200, mask_img)
 
-    assert out_img is not None
+    assert isinstance(out_img, nib.Nifti1Image)
 
 
 def test_make_local_connectivity_scorr():
@@ -81,7 +81,8 @@ def test_make_local_connectivity_scorr():
         f"smoothAROMAnonaggr_bold_short.nii.gz"
     func_img = nib.load(func_file)
     mask_img = nib.load(mask_file)
-    W = clustools.make_local_connectivity_scorr(func_img, mask_img, thresh=0.50)
+    W = clustools.make_local_connectivity_scorr(func_img, mask_img,
+                                                thresh=0.50)
 
     assert W is not None
 

diff --git a/tests/test_track.py b/tests/test_track.py
@@ -79,8 +79,8 @@ def test_prep_tissues(tiss_class):
     vent_csf_in_dwi = f"{base_dir}/003/dmri/csf_mask_dmri.nii.gz"
     wm_in_dwi = f"{base_dir}/003/dmri/wm_mask_dmri.nii.gz"
 
-    tiss_classifier = track.prep_tissues(t1w_mask, gm_in_dwi, vent_csf_in_dwi,
-                                         wm_in_dwi, tiss_class, B0_mask,
+    tiss_classifier = track.prep_tissues(nib.load(t1w_mask), nib.load(gm_in_dwi), nib.load(vent_csf_in_dwi),
+                                         nib.load(wm_in_dwi), tiss_class, nib.load(B0_mask),
                                          cmc_step_size=0.2)
     assert tiss_classifier is not None