mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
132 lines
4 KiB
Python
132 lines
4 KiB
Python
import pytest
|
|
import pickle
|
|
from epc_data.EpcClean import EpcClean
|
|
from pathlib import Path
|
|
|
|
# For local testing
|
|
if __file__ == "<input>":
|
|
input_data_path = Path("./epc_data/tests/test_data/EpcClean_inputs.obj")
|
|
else:
|
|
current_file_path = Path(__file__)
|
|
input_data_path = current_file_path.parent / 'test_data' / 'EpcClean_inputs.obj'
|
|
|
|
|
|
#
|
|
# @pytest.fixture
|
|
# def data():
|
|
# print("WOW")
|
|
# print(input_data_path)
|
|
# with open(input_data_path, 'rb') as f:
|
|
# data = pickle.load(f)
|
|
# return data
|
|
#
|
|
#
|
|
# def test_clean(data):
|
|
# epc = EpcClean(data)
|
|
# epc.clean()
|
|
# assert len(epc.cleaned["roof-description"]) == len(epc.unique_vals["roof-description"])
|
|
#
|
|
#
|
|
# def test_clean_roof(data):
|
|
# epc = EpcClean(data)
|
|
# result = epc.clean_roof('Pitched, 270 mm loft insulation')
|
|
#
|
|
# # change the expected output based on your requirement
|
|
# expected_output = {
|
|
# "is_pitched": True,
|
|
# "is_roof_room": False,
|
|
# "has_loft": True,
|
|
# "insulation_thickness": 270,
|
|
# "has_dwelling_above": False,
|
|
# "assumed": False,
|
|
# "is_flat": False,
|
|
# "thermal_transmittence": None,
|
|
# "thermal_transmittence_unit": None
|
|
# }
|
|
#
|
|
# assert result == expected_output
|
|
#
|
|
#
|
|
# def test_clean_roof_with_dwelling_above(data):
|
|
# epc = EpcClean(data)
|
|
# result = epc.clean_roof('(another dwelling above)')
|
|
#
|
|
# expected_output = {
|
|
# "is_pitched": False,
|
|
# "is_roof_room": False,
|
|
# "has_loft": False,
|
|
# "insulation_thickness": 0,
|
|
# "has_dwelling_above": True,
|
|
# "assumed": False,
|
|
# "is_flat": False,
|
|
# "is_thatched": False,
|
|
# "thermal_transmittence": None,
|
|
# "thermal_transmittence_unit": None,
|
|
# }
|
|
#
|
|
# assert result == expected_output
|
|
|
|
|
|
class TestEpcClean:
|
|
@pytest.fixture(autouse=True)
|
|
def load_data(self):
|
|
with open(input_data_path, "rb") as file:
|
|
self.data = pickle.load(file)
|
|
self.ec = EpcClean(self.data)
|
|
|
|
def test_clean(self):
|
|
self.ec.clean()
|
|
assert len(self.ec.cleaned["roof-description"]) == len(self.ec.unique_vals["roof-description"])
|
|
|
|
def test__init_empty_cleaned_obj(self):
|
|
self.ec._init_empty_cleaned_obj()
|
|
assert all([len(values) == 0 for values in self.ec.cleaned.values()])
|
|
|
|
def test__search_split_roof_description(self):
|
|
assert self.ec._search_split_roof_description("insulated") == "average"
|
|
assert self.ec._search_split_roof_description("limited") == "below average"
|
|
with pytest.raises(NotImplementedError):
|
|
self.ec._search_split_roof_description("unknown")
|
|
|
|
def test__find_insulation_thickness(self):
|
|
assert self.ec._find_insulation_thickness("no insulation", False, False, False) == 0
|
|
|
|
def test__extract_thermal_transmittence(self):
|
|
description = "U-value of 2.3 w/m-¦k"
|
|
assert self.ec._extract_thermal_transmittence(description) == (2.3, "w/m-¦k")
|
|
|
|
def test_clean_roof(self):
|
|
result = self.ec.clean_roof('Pitched, 270 mm loft insulation')
|
|
|
|
# change the expected output based on your requirement
|
|
expected_output = {
|
|
"is_pitched": True,
|
|
"is_roof_room": False,
|
|
"has_loft": True,
|
|
"insulation_thickness": 270,
|
|
"has_dwelling_above": False,
|
|
"assumed": False,
|
|
"is_flat": False,
|
|
"thermal_transmittence": None,
|
|
"thermal_transmittence_unit": None
|
|
}
|
|
|
|
assert result == expected_output
|
|
|
|
def test_clean_roof_with_dwelling_above(self):
|
|
result = self.ec.clean_roof('(another dwelling above)')
|
|
|
|
expected_output = {
|
|
"is_pitched": False,
|
|
"is_roof_room": False,
|
|
"has_loft": False,
|
|
"insulation_thickness": 0,
|
|
"has_dwelling_above": True,
|
|
"assumed": False,
|
|
"is_flat": False,
|
|
"is_thatched": False,
|
|
"thermal_transmittence": None,
|
|
"thermal_transmittence_unit": None,
|
|
}
|
|
|
|
assert result == expected_output
|