map elmhurst energy fields to epc property data class 🟥

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