Remove section_name From Flatahb If segment_code Is Empty, modified t…

…est (#341)
Hochfrequenz · Jul 9, 2024 · d42f8d8 · d42f8d8
1 parent b7d010d
commit d42f8d8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/kohlrahbi/unfoldedahb/unfoldedahbtable.py b/src/kohlrahbi/unfoldedahb/unfoldedahbtable.py
@@ -352,7 +352,7 @@ def convert_to_flat_ahb(self) -> FlatAnwendungshandbuch:
                     value_pool_entry=unfolded_ahb_line.code,
                     name=unfolded_ahb_line.beschreibung or unfolded_ahb_line.qualifier,
                     ahb_expression=unfolded_ahb_line.bedingung_ausdruck,
-                    section_name=unfolded_ahb_line.segment_name,
+                    section_name=unfolded_ahb_line.segment_name if unfolded_ahb_line.segment is not None else None,
                     index=unfolded_ahb_line.index,
                 )
             )

diff --git a/unittests/test_unfolded_ahb_table.py b/unittests/test_unfolded_ahb_table.py
@@ -111,7 +111,7 @@ def test_convert_to_flat_ahb(self) -> None:
             lines=[
                 AhbLine(
                     guid=None,
-                    section_name="Ansprechpartner",
+                    section_name=None,
                     segment_group_key="SG3",
                     segment_code=None,
                     data_element=None,
@@ -122,7 +122,7 @@ def test_convert_to_flat_ahb(self) -> None:
                 ),
                 AhbLine(
                     guid=None,
-                    section_name="Ansprechpartner",
+                    section_name=None,
                     segment_group_key="SG3",
                     segment_code=None,
                     data_element=None,