diff --git a/backend/documents_parser/tests/test_elmhurst_end_to_end.py b/backend/documents_parser/tests/test_elmhurst_end_to_end.py index 18742356..c434e833 100644 --- a/backend/documents_parser/tests/test_elmhurst_end_to_end.py +++ b/backend/documents_parser/tests/test_elmhurst_end_to_end.py @@ -5,18 +5,7 @@ from datetime import date import pytest from backend.documents_parser.elmhurst_extractor import ElmhurstSiteNotesExtractor -from datatypes.epc.domain.epc_property_data import ( - EpcPropertyData, - MainHeatingDetail, - SapBuildingPart, - SapEnergySource, - SapFloorDimension, - SapHeating, - SapVentilation, - SapWindow, - ShowerOutlet, - ShowerOutlets, -) +from datatypes.epc.domain.epc_property_data import EpcPropertyData from datatypes.epc.domain.mapper import EpcPropertyDataMapper FIXTURE_PATH = os.path.join( @@ -309,3 +298,20 @@ class TestDraughtproofingAndWater: def test_any_unheated_rooms_false(self, result: EpcPropertyData) -> None: assert result.any_unheated_rooms is False + + +class TestEnergyPerformance: + def test_energy_rating_current(self, result: EpcPropertyData) -> None: + assert result.energy_rating_current == 69 + + def test_energy_rating_potential(self, result: EpcPropertyData) -> None: + assert result.energy_rating_potential == 77 + + def test_environmental_impact_current(self, result: EpcPropertyData) -> None: + assert result.environmental_impact_current == 76 + + def test_environmental_impact_potential(self, result: EpcPropertyData) -> None: + assert result.environmental_impact_potential == 81 + + def test_co2_emissions_current(self, result: EpcPropertyData) -> None: + assert result.co2_emissions_current == 1.683