Add Test Case for Segment ID Parsing

unittests/test_ahb_sub_table.py

@@ -8,7 +8,8 @@
 from docx.table import Table
 
 from kohlrahbi.ahbtable.ahbsubtable import AhbSubTable
-from kohlrahbi.read_functions import get_all_paragraphs_and_tables
+from kohlrahbi.read_functions import get_ahb_table, get_all_paragraphs_and_tables
+from kohlrahbi.unfoldedahb import UnfoldedAhb
 
 
 class TestAhbSubTable:
@@ -41,3 +42,20 @@ def test_from_table(self) -> None:
             assert isinstance(ahb_sub_table, AhbSubTable)
         else:
             raise TypeError("You did not pass a docx table instance.")
+
+    def test_segment_id_parsing(self) -> None:
+        """
+        https://github.com/Hochfrequenz/kohlrahbi/issues/304
+        """
+        ahb_file_path: Path = Path(__file__).parent / Path(
+            "test-files/docx_files/UTILMDAHBStrom-informatorischeLesefassung1.2a_99991231_20240403.docx"
+        )
+        assert ahb_file_path.exists()
+        doc = docx.Document(str(ahb_file_path))  # Creating word reader object.
+        ahb_table = get_ahb_table(document=doc, pruefi="55109")
+        assert ahb_table is not None
+        unfolded_ahb = UnfoldedAhb.from_ahb_table(ahb_table=ahb_table, pruefi="55109")
+        assert unfolded_ahb is not None
+        flat_ahb = unfolded_ahb.convert_to_flat_ahb()
+        assert flat_ahb is not None
+        assert any(l for l in flat_ahb.lines if l.segment_id is not None)