From ecdd75c8e547ca2a9497c7dddfc9b2f25a615515 Mon Sep 17 00:00:00 2001 From: Cezar Moise Date: Wed, 18 Sep 2024 14:55:33 +0300 Subject: [PATCH] PR fixes --- tests/test_internal_functions.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/test_internal_functions.py b/tests/test_internal_functions.py index 76ca06fb..f69497e1 100644 --- a/tests/test_internal_functions.py +++ b/tests/test_internal_functions.py @@ -85,11 +85,8 @@ } ] -@pytest.mark.parametrize("test_case", test_cases, ids=[tc["id"] for tc in test_cases]) -def test_parse_tasks(test_case): - output = test_case["output"] - expected_tasks = test_case["expected_tasks"] - +@pytest.mark.parametrize("output, expected_tasks", [pytest.param(t["output"], t["expected_tasks"], id=t["id"]) for t in test_cases]) +def test_parse_tasks(output, expected_tasks): for ks, cf, expected in expected_tasks: n = Node._parse_tasks(output, ks, cf) assert n == expected, f"Expected {expected} tasks for {ks}.{cf}, but got {n}"