Adding tests

bclenet · Dec 12, 2024 · 4931792 · 4931792
1 parent 621117a
commit 4931792
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/tests/pipelines/test_team_DC61.py b/tests/pipelines/test_team_DC61.py
@@ -141,6 +141,31 @@ def test_confounds_file(temporary_data_dir):
     def test_group_level_contrasts():
         """ Test the get_group_contrasts method """
 
+        # Wrong parameter
+        assert PipelineTeamDC61.get_group_level_contrasts('wrong') is None
+
+        # effect_of_gain parameter
+        assert PipelineTeamDC61.get_group_level_contrasts('effect_of_gain') 
+
+        # effect_of_loss parameter
+        assert PipelineTeamDC61.get_group_level_contrasts('effect_of_loss') 
+        [
+                ['gain_param_range', 'T', ['equalIndifference', 'equalRange'], [0, 1]],
+                ['gain_param_indiff', 'T', ['equalIndifference', 'equalRange'], [1, 0]]
+        ]
+
+        if subject_level_contrast == 'effect_of_loss':
+            range_con = ['loss_param_range', 'T', ['equalIndifference', 'equalRange'], [0, 1]]
+            indiff_con = ['loss_param_indiff', 'T', ['equalIndifference', 'equalRange'], [1, 0]]
+            return [
+                ['loss_param_range_f', 'F', [range_con], [1]],
+                ['loss_param_indiff_f', 'F', [indiff_con], [1]]
+            ]
+
+
+
+
+
     @staticmethod
     @mark.unit_test
     def test_group_covariates(mock_participants_data):