def test_get_optional_artifacts_per_task_id(upstream_artifacts, expected): assert get_optional_artifacts_per_task_id(upstream_artifacts) == expected