From 71861770de17cc57228e542abe52fbea348094ce Mon Sep 17 00:00:00 2001
From: 36000 <jk6.28@outlook.com>
Date: Thu, 21 Sep 2023 13:16:58 -0700
Subject: [PATCH] more test updates

---
 AFQ/tests/test_api.py         | 4 ++--
 AFQ/tests/test_bundle_dict.py | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/AFQ/tests/test_api.py b/AFQ/tests/test_api.py
index 2fc08a284..c9d6400cc 100644
--- 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
index aadb95424..3bd9944f6 100644
--- 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