mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
14 lines
639 B
Python
14 lines
639 B
Python
from epc_data.tests.test_data.test_floor_attributes_cases import clean_floor_cases
|
|
from epc_data.attributes.FloorAttributes import FloorAttributes
|
|
|
|
|
|
class TestCleanFloor:
|
|
|
|
def test_clean_floor(self):
|
|
for test_case in clean_floor_cases:
|
|
result = FloorAttributes(test_case['original_description']).process()
|
|
# Ensure the output ordering is correct
|
|
expected_result = {key: test_case[key] for key in result.keys()}
|
|
expected_result["desc"] = test_case["original_description"]
|
|
result["desc"] = test_case["original_description"]
|
|
assert result == expected_result
|