junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

corrected unit tests:

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2024-01-19 17:25:53 +00:00
parent 80c35d42df
commit 43f3169e0c
2 changed files with 14 additions and 1009 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1000
backend/tests/test_sap_model_prep.py
View file

File diff suppressed because it is too large Load diff

23
etl/epc/tests/test_epcrecord.py
View file

@ -109,14 +109,13 @@ class TestEpcRecord:
assert record.prepared_epc["energy-consumption-current"] == 200.0
assert record.prepared_epc["co2-emissions-current"] == 5.5
def test_clean_energy_empty_values(self, cleaning_data, epc_records_1):
def test_clean_energy_empty_values(self, cleaning_data):
# We cannot have invalid values so this should raise an exception
record = EPCRecord(cleaning_data=cleaning_data)
record.prepared_epc = {
"energy-consumption-current": "",
"co2-emissions-current": ""
}
record._clean_energy()
with pytest.raises(ValueError):
record._clean_energy()
@ -207,7 +206,7 @@ class TestEpcRecord:
"mains-gas-flag": "InvalidValue"
}
# It should always be Y or N or an anomally value
with pytest.raises(ValueError):
with pytest.raises(KeyError):
record._clean_mains_gas()
record = EPCRecord(cleaning_data=cleaning_data)
@ -225,7 +224,8 @@ class TestEpcRecord:
}
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is True
assert record.prepared_epc["solar-water-heating-flag"] == "Y"
assert record.solar_water_heating_flag_bool is True
def test_clean_solar_hot_water_empty(self, cleaning_data):
record = EPCRecord(cleaning_data=cleaning_data)
@ -234,7 +234,8 @@ class TestEpcRecord:
}
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is None
assert record.prepared_epc["solar-water-heating-flag"] == "N"
assert record.solar_water_heating_flag_bool is False
def test_clean_number_lighting_outlets_valid(self, cleaning_data, epc_records_1):
record = EPCRecord(cleaning_data=cleaning_data, epc_records=epc_records_1)
@ -320,7 +321,8 @@ class TestEpcRecord:
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is True
assert record.prepared_epc["solar-water-heating-flag"] == "Y"
assert record.solar_water_heating_flag_bool is True
record = EPCRecord(cleaning_data=cleaning_data)
@ -330,7 +332,8 @@ class TestEpcRecord:
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is False
assert record.prepared_epc["solar-water-heating-flag"] == "N"
assert record.solar_water_heating_flag_bool is False
record = EPCRecord(cleaning_data=cleaning_data)
@ -340,7 +343,8 @@ class TestEpcRecord:
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is None
assert record.prepared_epc["solar-water-heating-flag"] == "N"
assert record.solar_water_heating_flag_bool is False
record = EPCRecord(cleaning_data=cleaning_data)
@ -350,4 +354,5 @@ class TestEpcRecord:
record._clean_solar_hot_water()
assert record.prepared_epc["solar-water-heating-flag"] is None
assert record.prepared_epc["solar-water-heating-flag"] == "N"
assert record.solar_water_heating_flag_bool is False