diff --git a/backend/documents_parser/tests/test_end_to_end.py b/backend/documents_parser/tests/test_end_to_end.py index 428d2cad..03c487fc 100644 --- a/backend/documents_parser/tests/test_end_to_end.py +++ b/backend/documents_parser/tests/test_end_to_end.py @@ -273,3 +273,22 @@ class TestPdfToEpcPropertyDataFixture2: def test_secondary_heating_type(self, result: EpcPropertyData) -> None: assert result.sap_heating.secondary_heating_type == "Open fire in grate" + + +PDF_PATH_3 = os.path.join( + os.path.dirname(__file__), "fixtures", "ExampleSiteNotes_3.pdf" +) + + +class TestPdfToEpcPropertyDataFixture3: + @pytest.fixture + def result(self) -> EpcPropertyData: + with open(PDF_PATH_3, "rb") as f: + pdf_bytes = f.read() + site_notes = PasHubRdSapSiteNotesExtractor( + pdf_to_text_list(pdf_bytes) + ).extract() + return EpcPropertyDataMapper.from_site_notes(site_notes) + + def test_immersion_heating_type(self, result: EpcPropertyData) -> None: + assert result.sap_heating.immersion_heating_type == "Dual"