@@ -47,25 +47,26 @@ def test_not_base_eligible(faker: Faker):
4747
4848
4949@pytest .mark .parametrize (
50- ("cohorts " , "test_comment" ),
50+ ("person_cohorts" , "iteration_cohorts " , "test_comment" ),
5151 [
52- (["elid_all_people" ], "Only magic cohort present" ),
53- (["elid_all_people" , "cohort1" ], "Magic cohort with other cohorts" ),
52+ (["cohort1" ], ["elid_all_people" ], "Only magic cohort present" ),
53+ (["cohort1" ], ["elid_all_people" , "cohort1" ], "Magic cohort with other cohorts" ),
54+ ([], ["elid_all_people" ], "Only magic cohort present without " ),
5455 ],
5556)
56- def test_base_eligible_with_when_magic_cohort_is_present (faker : Faker , cohorts , test_comment ):
57+ def test_base_eligible_with_when_magic_cohort_is_present (faker : Faker , person_cohorts , iteration_cohorts , test_comment ):
5758 # Given
5859 nhs_number = NHSNumber (faker .nhs_number ())
5960 date_of_birth = DateOfBirth (faker .date_of_birth (minimum_age = 76 , maximum_age = 79 ))
6061
61- person_rows = person_rows_builder (nhs_number , date_of_birth = date_of_birth , cohorts = [ "cohort1" ] )
62+ person_rows = person_rows_builder (nhs_number , date_of_birth = date_of_birth , cohorts = person_cohorts )
6263 campaign_configs = [
6364 rule_builder .CampaignConfigFactory .build (
6465 target = "RSV" ,
6566 iterations = [
6667 rule_builder .IterationFactory .build (
6768 iteration_cohorts = [
68- rule_builder .IterationCohortFactory .build (cohort_label = label ) for label in cohorts
69+ rule_builder .IterationCohortFactory .build (cohort_label = label ) for label in iteration_cohorts
6970 ],
7071 iteration_rules = [rule_builder .PersonAgeSuppressionRuleFactory .build ()],
7172 )
0 commit comments