From 1525fcfee9eb9fb93bed1fa9c41c053e8e06b10d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Boris=20Cl=C3=A9net?= Date: Wed, 31 Jan 2024 16:02:40 +0100 Subject: [PATCH] [TEST] test update for pipeline outputs --- tests/pipelines/test_team_U26C.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/tests/pipelines/test_team_U26C.py b/tests/pipelines/test_team_U26C.py index fac86525..93436ac0 100644 --- a/tests/pipelines/test_team_U26C.py +++ b/tests/pipelines/test_team_U26C.py @@ -72,19 +72,11 @@ def test_outputs(): pipeline = PipelineTeamU26C() # 1 - 1 subject outputs pipeline.subject_list = ['001'] - assert len(pipeline.get_preprocessing_outputs()) == 0 - assert len(pipeline.get_run_level_outputs()) == 0 - assert len(pipeline.get_subject_level_outputs()) == 7 - assert len(pipeline.get_group_level_outputs()) == 45 - assert len(pipeline.get_hypotheses_outputs()) == 18 + helpers.test_pipeline_outputs(pipeline, [0, 0, 7, 45, 18]) # 2 - 4 subjects outputs pipeline.subject_list = ['001', '002', '003', '004'] - assert len(pipeline.get_preprocessing_outputs()) == 0 - assert len(pipeline.get_run_level_outputs()) == 0 - assert len(pipeline.get_subject_level_outputs()) == 28 - assert len(pipeline.get_group_level_outputs()) == 45 - assert len(pipeline.get_hypotheses_outputs()) == 18 + helpers.test_pipeline_outputs(pipeline, [0, 0, 28, 45, 18]) @staticmethod @mark.unit_test