Test correaction

Inria-Empenn · Apr 9, 2024 · 8b1368a · 8b1368a
1 parent 333c54a
commit 8b1368a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/pipelines/test_team_3TR7.py b/tests/pipelines/test_team_3TR7.py
@@ -68,12 +68,12 @@ def test_subject_information():
         # Compare bunches to expected
         bunch = info[0]
         assert isinstance(bunch, Bunch)
-        assert bunch.conditions == ['gamble_run1']
+        assert bunch.conditions == ['gamble']
         helpers.compare_float_2d_arrays(bunch.onsets, [[4.071, 11.834, 19.535, 27.535, 36.435]])
         helpers.compare_float_2d_arrays(bunch.durations, [[4.0, 4.0, 4.0, 4.0, 4.0]])
         assert bunch.amplitudes is None
         assert bunch.tmod is None
-        assert bunch.pmod[0].name == ['gain_run1', 'loss_run1']
+        assert bunch.pmod[0].name == ['gain', 'loss']
         assert bunch.pmod[0].poly == [1, 1]
         helpers.compare_float_2d_arrays(bunch.pmod[0].param,
             [[14.0, 34.0, 38.0, 10.0, 16.0], [6.0, 14.0, 19.0, 15.0, 17.0]])
@@ -82,12 +82,12 @@ def test_subject_information():
 
         bunch = info[1]
         assert isinstance(bunch, Bunch)
-        assert bunch.conditions == ['gamble_run2']
+        assert bunch.conditions == ['gamble']
         helpers.compare_float_2d_arrays(bunch.onsets, [[4.071, 11.834, 19.535, 27.535, 36.435]])
         helpers.compare_float_2d_arrays(bunch.durations, [[4.0, 4.0, 4.0, 4.0, 4.0]])
         assert bunch.amplitudes is None
         assert bunch.tmod is None
-        assert bunch.pmod[0].name == ['gain_run2', 'loss_run2']
+        assert bunch.pmod[0].name == ['gain', 'loss']
         assert bunch.pmod[0].poly == [1, 1]
         helpers.compare_float_2d_arrays(bunch.pmod[0].param,
             [[14.0, 34.0, 38.0, 10.0, 16.0], [6.0, 14.0, 19.0, 15.0, 17.0]])