diff --git a/backend/documents_parser/tests/test_end_to_end.py b/backend/documents_parser/tests/test_end_to_end.py index 1310859f..f3ca2786 100644 --- a/backend/documents_parser/tests/test_end_to_end.py +++ b/backend/documents_parser/tests/test_end_to_end.py @@ -20,6 +20,7 @@ from datatypes.epc.domain.epc_property_data import ( from datatypes.epc.domain.mapper import EpcPropertyDataMapper PDF_PATH = os.path.join(os.path.dirname(__file__), "fixtures", "ExampleSiteNotes.pdf") +PDF_PATH_2 = os.path.join(os.path.dirname(__file__), "fixtures", "ExampleSiteNotes_2.pdf") class TestPdfToEpcPropertyData: @@ -246,3 +247,15 @@ class TestPdfToEpcPropertyData: postcode="CW1 4JR", report_reference="6EA2A86D-94CE-4792-8D49-AB495C744EDD", ) + + +class TestPdfToEpcPropertyDataFixture2: + @pytest.fixture + def result(self) -> EpcPropertyData: + with open(PDF_PATH_2, "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_cylinder_insulation_thickness(self, result: EpcPropertyData) -> None: + assert result.sap_heating.cylinder_insulation_thickness == 38