more test updates

yeatmanlab · Sep 21, 2023 · 7186177 · 7186177
1 parent 8da0d0b
commit 7186177
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/AFQ/tests/test_api.py b/AFQ/tests/test_api.py
@@ -528,7 +528,7 @@ def test_AFQ_slr():
     afd.read_stanford_hardi_tractography()
 
     _, bids_path, _ = get_temp_hardi()
-    bd = BundleDict(["CST_L"])
+    bd = BundleDict(["CST_L", "CST_R"])
 
     myafq = GroupAFQ(
         bids_path=bids_path,
@@ -845,7 +845,7 @@ def test_AFQ_data_waypoint():
     assert op.exists(op.join(
         myafq.export("results_dir"),
         'bundles',
-        'sub-01_ses-01_coordsys-RASMM_trkmodel-probCSD_algo-AFQ_desc-SLFL_tractography.trk'))  # noqa
+        'sub-01_ses-01_coordsys-RASMM_trkmodel-probCSD_bundlerec-AFQ_desc-SLFL_tractography.trk'))  # noqa
 
     tract_profile_fname = myafq.export("profiles")
     tract_profiles = pd.read_csv(tract_profile_fname)

diff --git a/AFQ/tests/test_bundle_dict.py b/AFQ/tests/test_bundle_dict.py
@@ -29,10 +29,10 @@ def test_BundleDict():
 
     assert len(afq_bundles) == 2
 
-    # Forceps Minor
-    afq_bundles = abd.BundleDict(["FA"])
+    # Forceps Minor and Major
+    afq_bundles = abd.BundleDict(["FA", "FP"])
 
-    assert len(afq_bundles) == 1
+    assert len(afq_bundles) == 2
 
     # Cingulum Hippocampus
     # not included but exists in templates