From 5ab48f3bb4828c2c7edbc45a682146dbe8ae321f Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Fri, 27 Mar 2026 00:52:13 +0000 Subject: [PATCH] Fixing optimiser tests --- recommendations/tests/test_optimiser_functions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/recommendations/tests/test_optimiser_functions.py b/recommendations/tests/test_optimiser_functions.py index 40fa56b6..3588877f 100644 --- a/recommendations/tests/test_optimiser_functions.py +++ b/recommendations/tests/test_optimiser_functions.py @@ -99,7 +99,8 @@ class TestCalculateGain: def test_returns_zero_for_already_installed_getting_to_target(self): body = SimpleNamespace(goal="Increasing EPC", goal_value="C") - p = SimpleNamespace(data={"current-energy-efficiency": "67"}, id=1) + epc_record = SimpleNamespace(current_energy_efficiency=67) + p = SimpleNamespace(epc_record=epc_record, id=1) fixed_gain = 0 eco_packages = {1: (None, None, None, [])} already_installed_sap = 2 @@ -107,7 +108,6 @@ class TestCalculateGain: body=body, p=p, fixed_gain=fixed_gain, - eco_packages=eco_packages, already_installed_gain=already_installed_sap )