mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
113 lines
5.3 KiB
Python
113 lines
5.3 KiB
Python
import pytest
|
|
from recommendations.SolarPvRecommendations import SolarPvRecommendations
|
|
from backend.Property import Property
|
|
from etl.epc.Record import EPCRecord
|
|
import pandas as pd
|
|
from datetime import datetime
|
|
from utils.s3 import read_dataframe_from_s3_parquet, read_from_s3
|
|
from etl.solar.SolarPhotoSupply import SolarPhotoSupply
|
|
from recommendations.Recommendations import Recommendations
|
|
from backend.ml_models.api import ModelApi
|
|
import msgpack
|
|
|
|
|
|
class TestSolarPvRecommendations:
|
|
@pytest.fixture
|
|
def property_instance_invalid_type(self):
|
|
# Setup the property_instance with an invalid property type
|
|
epc_record = EPCRecord()
|
|
epc_record.prepared_epc = {
|
|
"property-type": "InvalidType", "county": "Broxbourne", "photo-supply": None
|
|
}
|
|
property_instance_invalid_type = Property(id=1, address="", postcode="", epc_record=epc_record)
|
|
property_instance_invalid_type.roof = {"is_flat": False, "is_pitched": False, "is_roof_room": False}
|
|
return property_instance_invalid_type
|
|
|
|
@pytest.fixture
|
|
def property_instance_invalid_roof(self):
|
|
# Setup the property_instance with invalid roof type
|
|
epc_record = EPCRecord()
|
|
epc_record.prepared_epc = {
|
|
"county": "Huntingdonshire", "property-type": "House", "photo-supply": None
|
|
}
|
|
property_instance_invalid_roof = Property(id=1, address="", postcode="", epc_record=epc_record)
|
|
property_instance_invalid_roof.roof = {"is_flat": False, "is_pitched": False, "is_roof_room": False}
|
|
return property_instance_invalid_roof
|
|
|
|
@pytest.fixture
|
|
def property_instance_has_solar_pv(self):
|
|
# Setup the property_instance without existing solar pv
|
|
epc_record = EPCRecord()
|
|
epc_record.prepared_epc = {"photo-supply": "40", "county": "Huntingdonshire",
|
|
"property-type": "House"}
|
|
property_instance_has_solar_pv = Property(id=1, address="", postcode="", epc_record=epc_record)
|
|
property_instance_has_solar_pv.roof = {"is_flat": True}
|
|
return property_instance_has_solar_pv
|
|
|
|
@pytest.fixture
|
|
def property_instance_valid_all(self):
|
|
# Setup a valid property_instance that passes all conditions
|
|
epc_record = EPCRecord()
|
|
epc_record.prepared_epc = {"property-type": "House", "photo-supply": None, "county": "Huntingdonshire"}
|
|
property_instance_valid_all = Property(id=1, address="", postcode="", epc_record=epc_record)
|
|
property_instance_valid_all.roof_area = 40
|
|
property_instance_valid_all.number_of_floors = 2
|
|
property_instance_valid_all.roof = {"is_flat": True}
|
|
property_instance_valid_all.solar_panel_configuration = {
|
|
"panel_performance": pd.DataFrame(
|
|
[
|
|
{
|
|
"panneled_roof_area": 20,
|
|
"n_panels": 10,
|
|
"array_wattage": 4000,
|
|
"initial_ac_kwh_per_year": 3800
|
|
}
|
|
]
|
|
)
|
|
}
|
|
|
|
return property_instance_valid_all
|
|
|
|
def test_invalid_property_type(self, property_instance_invalid_type):
|
|
solar_pv = SolarPvRecommendations(property_instance_invalid_type)
|
|
solar_pv.recommend(phase=0)
|
|
assert not solar_pv.recommendation
|
|
|
|
def test_invalid_roof_type(self, property_instance_invalid_roof):
|
|
solar_pv = SolarPvRecommendations(property_instance_invalid_roof)
|
|
solar_pv.recommend(phase=0)
|
|
assert not solar_pv.recommendation
|
|
|
|
def test_existing_solar_pv(self, property_instance_has_solar_pv):
|
|
solar_pv = SolarPvRecommendations(property_instance_has_solar_pv)
|
|
solar_pv.recommend(phase=0)
|
|
assert not solar_pv.recommendation
|
|
|
|
def test_valid_all_conditions(self, property_instance_valid_all):
|
|
solar_pv = SolarPvRecommendations(property_instance_valid_all)
|
|
solar_pv.recommend(phase=0)
|
|
assert solar_pv.recommendation == [
|
|
{
|
|
'phase': 0, 'parts': [], 'type': 'solar_pv',
|
|
'description': 'Install a 4.0 kilowatt-peak (kWp) solar photovoltaic (PV) panel system on 50% the '
|
|
'roof.',
|
|
'starting_u_value': None, 'new_u_value': None, 'sap_points': None, 'already_installed': False,
|
|
'total': 4850.0, 'subtotal': 4041.666666666667, 'vat': 808.333333333333, 'labour_hours': 48,
|
|
'labour_days': 2, 'photo_supply': 50.0, 'has_battery': False, 'initial_ac_kwh_per_year': 3800,
|
|
'description_simulation': {'photo-supply': 50.0}
|
|
},
|
|
{
|
|
'phase': 0, 'parts': [], 'type': 'solar_pv',
|
|
'description': 'Install a 4.0 kilowatt-peak (kWp) '
|
|
'solar photovoltaic (PV) panel system '
|
|
'on 50% the roof, with a battery '
|
|
'storage system.',
|
|
'starting_u_value': None, 'new_u_value': None,
|
|
'sap_points': None, 'already_installed': False,
|
|
'total': 7550.0, 'subtotal': 6291.666666666667,
|
|
'vat': 1258.333333333333, 'labour_hours': 48,
|
|
'labour_days': 2, 'photo_supply': 50.0,
|
|
'has_battery': True, 'initial_ac_kwh_per_year': 3800,
|
|
'description_simulation': {'photo-supply': 50.0}
|
|
}
|
|
]
|