diff --git a/recommendations/tests/test_recommendation_utils.py b/recommendations/tests/test_recommendation_utils.py index 73796979..69f0d1b6 100644 --- a/recommendations/tests/test_recommendation_utils.py +++ b/recommendations/tests/test_recommendation_utils.py @@ -42,10 +42,12 @@ class TestRecommendationUtils: assert recommendation_utils.update_lowest_selected_u_value(1, 0.5) == 0.5 def test_get_recommended_part(self): - part = {'depths': [1, 2, 3]} + part = {'description': "some insulation material"} + 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} + part=part, cost_result={"cost_result": 123}, quantity=99, quantity_unit="m2" + ) == {'description': "some insulation material", 'quantity': 99, 'quantity_unit': QuantityUnits.m2.value, + "cost_result": 123} def test_get_roof_u_value(self): # Test case 1: Insulation thickness is known and is_loft is True