From eae1a5538a2a83fd543d15c8ed58f164c7aac118 Mon Sep 17 00:00:00 2001 From: 36000 Date: Thu, 6 Jun 2024 12:11:54 -0700 Subject: [PATCH] update test to fixed orientation system --- AFQ/recognition/tests/test_other_bundles.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/AFQ/recognition/tests/test_other_bundles.py b/AFQ/recognition/tests/test_other_bundles.py index e68caae9..7562afa8 100644 --- a/AFQ/recognition/tests/test_other_bundles.py +++ b/AFQ/recognition/tests/test_other_bundles.py @@ -36,7 +36,7 @@ def test_clean_relative_to_other_core(): assert isinstance(cleaned_idx_core, np.ndarray) assert cleaned_idx_core.shape[0] == this_bundle_sls_sample.shape[0] - if core == "inferior": - assert np.all(cleaned_idx_core == [False, True]) + if core == "inferior" or core == "right": + assert np.all(cleaned_idx_core == [True, True]) else: assert np.all(cleaned_idx_core == [False, False])