mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
import pickle
|
|
import pytest
|
|
import os
|
|
from unittest.mock import Mock
|
|
from model_data.recommendations.FloorRecommendations import FloorRecommendations
|
|
|
|
|
|
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
|