Skip to content

Commit f0ce5f7

Browse files
Added addtional test case.
1 parent 6da40a3 commit f0ce5f7

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

tests/unit/services/calculators/test_eligibility_calculator.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)