handle some plans having zero cost 🟥

This commit is contained in:
Daniel Roth 2026-02-20 11:24:17 +00:00
parent b4583d3c8b
commit 678de56def

View file

@ -118,6 +118,27 @@ def test_all_plans_zero_cost__highest_priority_returned(
assert actual_default_plan.id == expected_default_plan_id
def test_some_plans_zero_cost__cheapest_returned(
created_at_datetime: datetime,
) -> None:
# arrange
epc_c_plan = make_plan(created_at_datetime, True, cost_of_works=0.0, name="EPC C")
minor_works_plan = make_plan(
created_at_datetime, False, cost_of_works=50.0, name="EPC C - Minor Works"
)
scenario_priority_order: List[int] = [4, 3]
expected_default_plan_id = 2
# act
actual_default_plan = choose_cheapest_relevant_plan(
plans=[epc_c_plan, minor_works_plan],
scenario_priority_order=scenario_priority_order,
)
# assert
assert actual_default_plan.id == expected_default_plan_id
def test_all_plans_null_cost__highest_priority_returned(
created_at_datetime: datetime,
) -> None: