mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
import pickle
|
|
import pytest
|
|
import os
|
|
from unittest.mock import Mock
|
|
from model_data.recommendations.FloorRecommendations import FloorRecommendations
|
|
|
|
|
|
# with open(
|
|
# os.path.abspath(os.path.dirname(__file__)) + "/model_data/tests/test_data/input_properties.pkl", "rb"
|
|
# ) as f:
|
|
# input_properties = pickle.load(f)
|
|
#
|
|
# with open(
|
|
# os.path.abspath(os.path.dirname(__file__)) + "/model_data/tests/test_data/uvalue_estimates.pkl", "rb"
|
|
# ) as f:
|
|
# uvalue_estimates = pickle.load(f)
|
|
|
|
|
|
class TestWallRecommendations:
|
|
|
|
@pytest.fixture
|
|
def input_properties(self):
|
|
with open(
|
|
os.path.abspath(os.path.dirname(__file__)) + "/test_data/input_properties.pkl", "rb"
|
|
) as f:
|
|
return pickle.load(f)
|
|
|
|
@pytest.fixture
|
|
def uvalue_estimates(self):
|
|
with open(
|
|
os.path.abspath(os.path.dirname(__file__)) + "/test_data/uvalue_estimates.pkl", "rb"
|
|
) as f:
|
|
return pickle.load(f)
|
|
|
|
@pytest.fixture
|
|
def mock_floor_rec_instance(self):
|
|
# Creating a mock instance of WallRecommendations with the necessary attributes
|
|
property_mock = Mock()
|
|
property_mock.full_sap_epc = {"lodgement-date": "2000-01-01"} # or any date you want
|
|
property_mock.data = {"construction-age-band": "1950"} # or any other data that fits your tests
|
|
|
|
uvalue_estimates_mock = Mock()
|
|
|
|
mock_wall_rec_instance = FloorRecommendations(property_mock, uvalue_estimates_mock)
|
|
return mock_wall_rec_instance
|
|
|
|
def test_init(self, input_properties, uvalue_estimates):
|
|
obj = FloorRecommendations(property_instance=input_properties[0], uvalue_estimates=uvalue_estimates)
|
|
assert obj
|
|
assert obj.property
|
|
assert obj.uvalue_estimates
|
|
|
|
def test_other_premises_below(self, input_properties, uvalue_estimates):
|
|
recommender = FloorRecommendations(property_instance=input_properties[0], uvalue_estimates=uvalue_estimates)
|
|
recommender.recommend()
|
|
assert recommender.property.floor["another_property_below"]
|
|
|
|
assert not recommender.recommendations
|
|
|
|
def test_suspended_no_insulation(self, input_properties, uvalue_estimates):
|
|
"""
|
|
For a suspended floor without insulation, we use the rdsap methogology to estimate a U-value for the floor
|
|
:return:
|
|
"""
|
|
recommender = FloorRecommendations(property_instance=input_properties[2], uvalue_estimates=uvalue_estimates)
|
|
recommender.recommend()
|
|
assert recommender.property.floor["is_suspended"]
|
|
assert recommended.es
|
|
assert recommender.recommendations
|