Skip to content

Commit b4f59d1

Browse files
test fixes and lint fixes
1 parent 27a3ce0 commit b4f59d1

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/eligibility_signposting_api/model/rules.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
StartDate = NewType("StartDate", date)
3030
EndDate = NewType("EndDate", date)
3131
CohortLabel = NewType("CohortLabel", str)
32+
RuleStop = NewType("RuleStop", bool)
3233

3334

3435
class RuleType(StrEnum):
@@ -99,7 +100,7 @@ class IterationRule(BaseModel):
99100
operator: RuleOperator = Field(..., alias="Operator")
100101
comparator: RuleComparator = Field(..., alias="Comparator")
101102
attribute_target: RuleAttributeTarget | None = Field(None, alias="AttributeTarget")
102-
rule_stop: bool = Field(default=False, alias="RuleStop")
103+
rule_stop: RuleStop | None = Field(None, alias="RuleStop")
103104

104105
@field_validator("rule_stop", mode="before")
105106
def parse_yn_to_bool(cls, v: str) -> bool: # noqa: N805

tests/unit/services/calculators/test_eligibility_calculator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -847,8 +847,6 @@ def test_rules_stop_behavior(rule_stop: str | None, expected_status: Status, tes
847847
{"Type": "F", "Priority": 15},
848848
]
849849

850-
iteration_rules = [IterationRule.model_validate({**simple_age_data, **variant}) for variant in rule_variants]
851-
852850
# Build campaign configuration
853851
campaign_config = rule_builder.CampaignConfigFactory.build(
854852
target="RSV",
@@ -858,7 +856,9 @@ def test_rules_stop_behavior(rule_stop: str | None, expected_status: Status, tes
858856
)
859857
],
860858
)
861-
campaign_config.iterations[0].iteration_rules.extend(iteration_rules)
859+
campaign_config.iterations[0].iteration_rules.extend(
860+
[IterationRule.model_validate({**simple_age_data, **variant}) for variant in rule_variants]
861+
)
862862

863863
calculator = EligibilityCalculator(person_rows, [campaign_config])
864864

0 commit comments

Comments
 (0)