mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
52 lines
2.5 KiB
Python
52 lines
2.5 KiB
Python
import pytest
|
|
from etl.epc.Record import EPCRecord
|
|
from backend.Property import Property
|
|
from recommendations.LightingRecommendations import LightingRecommendations
|
|
|
|
from recommendations.tests.test_data.materials import materials
|
|
|
|
|
|
class TestLightingRecommendations:
|
|
|
|
def test_init_invalid_materials(self):
|
|
epc_record = EPCRecord()
|
|
epc_record.county = "Greater London Authority"
|
|
input_property0 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record)
|
|
input_property0.lighting = {"low_energy_proportion": 0}
|
|
input_property0.already_installed = []
|
|
# Test for invalid materials
|
|
with pytest.raises(ValueError):
|
|
LightingRecommendations(input_property0, [])
|
|
|
|
def test_recommend_no_action_needed(self):
|
|
# Case where no recommendation is needed
|
|
epc_record = EPCRecord()
|
|
epc_record.county = "Greater London Authority"
|
|
input_property1 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record)
|
|
input_property1.lighting = {"low_energy_proportion": 100}
|
|
input_property1.already_installed = []
|
|
|
|
lr = LightingRecommendations(input_property1, materials)
|
|
lr.recommend()
|
|
assert lr.recommendation == []
|
|
|
|
def test_recommend_action_needed(self):
|
|
# Case where recommendation is needed
|
|
epc_record = EPCRecord()
|
|
epc_record.county = "Greater London Authority"
|
|
input_property1 = Property(id=1, postcode="F4k3 6", address="623 fake street", epc_record=epc_record)
|
|
input_property1.lighting = {"low_energy_proportion": 0.80}
|
|
input_property1.number_lighting_outlets = 20
|
|
input_property1.already_installed = []
|
|
|
|
lr = LightingRecommendations(input_property1, materials)
|
|
lr.recommend()
|
|
assert len(lr.recommendation) == 1
|
|
|
|
# Note - this test may be dependent on the ofgem price caps
|
|
assert lr.recommendation[0]["description_simulation"] == {'lighting-energy-eff': 'Very Good',
|
|
'lighting-description': 'Low energy lighting in all '
|
|
'fixed outlets',
|
|
'low-energy-lighting': 100}
|
|
assert lr.recommendation[0]["description"] == 'Install low energy lighting in 4 outlets'
|
|
assert lr.recommendation[0]["total"] == 14
|