Merge pull request #566 from 36000/fix_slr

[FIX] Put tg in rasmm first for SLR registration
yeatmanlab · Oct 29, 2020 · 9aead57 · 9aead57
2 parents b17c84f + f22c852
commit 9aead57
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/AFQ/api.py b/AFQ/api.py
@@ -862,6 +862,7 @@ def _reg_img(self, img, mask, row=None):
                                      img,
                                      Space.VOX,
                                      bbox_valid_check=False)
+                tg.to_rasmm()
                 return img, tg.streamlines
             elif img_l == "hcp_atlas":
                 atlas_fname = op.join(

diff --git a/AFQ/tests/test_api.py b/AFQ/tests/test_api.py
@@ -331,7 +331,7 @@ def test_API_type_checking():
         api.AFQ(bids_path, bundle_info=[2, 3])
 
 
-@pytest.mark.skip(reason="may cause OOM")
+@pytest.mark.nightly5
 def test_AFQ_slr():
     """
     Test if API can run using slr map
@@ -342,7 +342,10 @@ def test_AFQ_slr():
         dmriprep='vistasoft',
         reg_subject='subject_sls',
         reg_template='hcp_atlas')
-    myafq.export_rois()
+
+    tgram = load_tractogram(myafq.get_clean_bundles()[0], myafq.dwi_img[0])
+    bundles = aus.tgram_to_bundles(tgram, myafq.bundle_dict, myafq.dwi_img[0])
+    npt.assert_(len(bundles['CST_L']) > 0)
 
 
 @pytest.mark.nightly2