import pytest from unittest.mock import MagicMock from recommendations import recommendation_utils from datatypes.enums import QuantityUnits from recommendations.tests.test_data.wall_uvalue_test_cases import wall_uvalue_test_cases class TestRecommendationUtils: @pytest.fixture def property_mock(self): PropertyMock = MagicMock() PropertyMock.data = { 'total_floor_area_group_decile': 'Decile 1', 'number-habitable-rooms': 3, 'number-heated-rooms': 2 } return PropertyMock def test_r_value_per_mm_to_u_value(self): assert recommendation_utils.r_value_per_mm_to_u_value(1, 2) == 0.5 with pytest.raises(ZeroDivisionError): recommendation_utils.r_value_per_mm_to_u_value(0, 2) def test_calculate_u_value_uplift(self): assert recommendation_utils.calculate_u_value_uplift(1, 2) == (0.33333333333333337, 0.6666666666666666) with pytest.raises(ZeroDivisionError): recommendation_utils.calculate_u_value_uplift(0, 2) with pytest.raises(ZeroDivisionError): recommendation_utils.calculate_u_value_uplift(1, 0) def test_is_diminishing_returns(self): assert not recommendation_utils.is_diminishing_returns([1, 2, 3], 1, 1, 1) assert recommendation_utils.is_diminishing_returns([1, 2, 3], 0.5, 1, 1) assert not recommendation_utils.is_diminishing_returns([], 1, None, 1) def test_update_lowest_selected_u_value(self): assert recommendation_utils.update_lowest_selected_u_value(1, 2) == 1 assert recommendation_utils.update_lowest_selected_u_value(None, 2) == 2 assert recommendation_utils.update_lowest_selected_u_value(1, 0.5) == 0.5 def test_get_recommended_part(self): part = {'depths': [1, 2, 3]} assert recommendation_utils.get_recommended_part( part=part, selected_depth=1, selected_total_cost=50, quantity=99, quantity_unit="m2" ) == {'depths': [1], 'estimated_cost': 50, 'quantity': 99, 'quantity_unit': QuantityUnits.m2.value} def test_get_roof_u_value(self): # Test case 1: Insulation thickness is known and is_loft is True inputs = { 'insulation_thickness': '50', 'is_loft': True, 'is_roof_room': False, 'is_thatched': False, 'has_dwelling_above': False, 'is_flat': False, 'is_pitched': True, 'is_at_rafters': False, } for age_band in ["A", "B", "C", "D"]: assert recommendation_utils.get_roof_u_value(**{**inputs, "age_band": age_band}) == 0.68 def test_get_roof_u_value_case_2(self): inputs = { 'original_description': 'Pitched, 400+ mm insulation at joists', 'clean_description': 'Pitched, 400+ mm insulation at joists', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': True, 'is_roof_room': False, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': '400+', 'age_band': "J" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.16, f"Expected 0.16, but got {u_value}" def test_get_roof_u_value_case_3(self): inputs = { 'original_description': 'Room-in-roof, 200 mm insulation at rafters', 'clean_description': 'Room-in-roof, 200 mm insulation at rafters', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': False, 'is_roof_room': True, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': True, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': '200', 'age_band': "J" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.21, f"Expected 0.21, but got {u_value}" def test_get_roof_u_value_case_4(self): inputs = { 'original_description': 'Pitched, below average insulation', 'clean_description': 'Pitched, below average insulation', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': True, 'is_roof_room': False, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': 'below average', 'age_band': "E" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 1.5, f"Expected 1.5, but got {u_value}" def test_get_roof_u_value_case_5(self): # Test case where insulation thickness is exactly specified inputs = { 'original_description': 'Pitched, 100mm insulation', 'clean_description': 'Pitched, 100mm insulation', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': True, 'is_roof_room': False, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': '100', 'age_band': "G" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.40, f"Expected 0.40, but got {u_value}" def test_get_roof_u_value_case_6(self): # Test case for a thatched roof inputs = { 'original_description': 'Thatched, 75mm insulation', 'clean_description': 'Thatched, 75mm insulation', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': False, 'is_roof_room': False, 'is_loft': False, 'is_flat': False, 'is_thatched': True, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': '75', 'age_band': "H" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.35, f"Expected 0.35, but got {u_value}" def test_get_roof_u_value_case_7(self): # Test case where the roof has a room in it inputs = { 'original_description': 'Pitched, room-in-roof, 100mm insulation', 'clean_description': 'Pitched, room-in-roof, 100mm insulation', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': True, 'is_roof_room': True, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': False, 'is_valid': True, 'insulation_thickness': '100', 'age_band': "J" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.40, f"Expected 0.40, but got {u_value}" def test_get_roof_u_value_case_8(self): # Test case where there is a dwelling above the roof, U-value should be 0 inputs = { 'original_description': 'Pitched, 100mm insulation', 'clean_description': 'Pitched, 100mm insulation', 'thermal_transmittance': None, 'thermal_transmittance_unit': None, 'is_pitched': True, 'is_roof_room': False, 'is_loft': False, 'is_flat': False, 'is_thatched': False, 'is_at_rafters': False, 'is_assumed': False, 'has_dwelling_above': True, 'is_valid': True, 'insulation_thickness': '100', 'age_band': "J" } u_value = recommendation_utils.get_roof_u_value(**inputs) assert u_value == 0.0, f"Expected 0.0, but got {u_value}" @pytest.mark.parametrize( "test_case", wall_uvalue_test_cases ) def test_get_wall_uvalue(self, test_case): expected_uvalue = test_case["uvalue"] inputs = test_case.copy() del inputs["uvalue"] uvalue = recommendation_utils.get_wall_u_value(**inputs) assert expected_uvalue == uvalue, f"Expected u value {expected_uvalue}, recieved {uvalue}"