diff --git a/recommendations/tests/test_ventilation_recommendations.py b/recommendations/tests/test_ventilation_recommendations.py index ea87a632..15c9435c 100644 --- a/recommendations/tests/test_ventilation_recommendations.py +++ b/recommendations/tests/test_ventilation_recommendations.py @@ -10,6 +10,7 @@ class TestVentilationRecommendations: epc_record = EPCRecord() epc_record.prepared_epc = {"mechanical-ventilation": "natural"} input_property1 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record) + input_property1.already_installed = [] recommender = VentilationRecommendations( property_instance=input_property1, @@ -22,16 +23,18 @@ class TestVentilationRecommendations: assert len(recommender.recommendation) == 1 - assert recommender.recommendation[0]["total"] == 1071.0 + assert recommender.recommendation[0]["total"] == 560.0 assert recommender.recommendation[0]["type"] == "mechanical_ventilation" assert len(recommender.recommendation[0]["parts"]) == 1 - assert recommender.recommendation[0]["parts"][0]["description"] == 'Mechanical Extract Ventilation' + assert recommender.recommendation[0]["parts"][0][ + "description"] == 'Decentralised mechanical extract ventilation' assert recommender.recommendation[0]["parts"][0]["quantity"] == 2 def test_missing_ventilation(self): epc_record = EPCRecord() epc_record.prepared_epc = {"mechanical-ventilation": None} input_property2 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record) + input_property2.already_installed = [] recommender2 = VentilationRecommendations( property_instance=input_property2, @@ -44,16 +47,18 @@ class TestVentilationRecommendations: assert len(recommender2.recommendation) == 1 - assert recommender2.recommendation[0]["total"] == 1071.0 + assert recommender2.recommendation[0]["total"] == 560.0 assert recommender2.recommendation[0]["type"] == "mechanical_ventilation" assert len(recommender2.recommendation[0]["parts"]) == 1 - assert recommender2.recommendation[0]["parts"][0]["description"] == 'Mechanical Extract Ventilation' + assert recommender2.recommendation[0]["parts"][0][ + "description"] == 'Decentralised mechanical extract ventilation' assert recommender2.recommendation[0]["parts"][0]["quantity"] == 2 def test_nodata_ventilation(self): epc_record = EPCRecord() epc_record.prepared_epc = {"mechanical-ventilation": "NO DATA!!"} input_property3 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record) + input_property3.already_installed = [] recommender3 = VentilationRecommendations( property_instance=input_property3, @@ -66,16 +71,18 @@ class TestVentilationRecommendations: assert len(recommender3.recommendation) == 1 - assert recommender3.recommendation[0]["total"] == 1071.0 + assert recommender3.recommendation[0]["total"] == 560.0 assert recommender3.recommendation[0]["type"] == "mechanical_ventilation" assert len(recommender3.recommendation[0]["parts"]) == 1 - assert recommender3.recommendation[0]["parts"][0]["description"] == 'Mechanical Extract Ventilation' + assert recommender3.recommendation[0]["parts"][0][ + "description"] == 'Decentralised mechanical extract ventilation' assert recommender3.recommendation[0]["parts"][0]["quantity"] == 2 def test_existing_ventilation_1(self): epc_record = EPCRecord() epc_record.prepared_epc = {"mechanical-ventilation": "mechanical, extract only"} input_property4 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record) + input_property4.already_installed = [] input_property4.identify_ventilation() assert input_property4.has_ventilation @@ -94,6 +101,7 @@ class TestVentilationRecommendations: epc_record = EPCRecord() epc_record.prepared_epc = {"mechanical-ventilation": "mechanical, supply and extract"} input_property5 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record) + input_property5.already_installed = [] input_property5.identify_ventilation() assert input_property5.has_ventilation