fix: Apply suggestions from code review

Co-authored-by: Ernst Würger <[email protected]>
DSD-DBS · Nov 20, 2024 · f31a258 · f31a258
1 parent 673fc9b
commit f31a258
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/capellambse_context_diagrams/context.py b/capellambse_context_diagrams/context.py
@@ -536,7 +536,6 @@ def _create_edge(
         src_id, trg_id = port_id, interface_id
         routing_points = [port_middle, interface_middle]
     else:
-        # styleclass == "FOP"
         eid = f"__PortOutputAllocation:{port_id}"
         src_id, trg_id = interface_id, port_id
         routing_points = [interface_middle, port_middle]

diff --git a/tests/test_cable_tree_views.py b/tests/test_cable_tree_views.py
@@ -15,16 +15,16 @@
             ("39e96ffc-2f32-41b9-b406-ba82c78fe451", 8), id="Inside Tree"
         ),
         pytest.param(
-            ("6c607b75-504a-4d68-966b-0982fde3275e", 20), id="Outside Tree"
+            ("6c607b75-504a-4d68-966b-0982fde3275e", 14), id="Outside Tree"
         ),
     ],
 )
 def test_cable_tree_views(
     model: capellambse.MelodyModel, diagram_elements: tuple[str, int]
 ) -> None:
-    uuid, elements_n = diagram_elements
+    uuid, number_of_elements = diagram_elements
     obj = model.by_uuid(uuid)
 
     diag = obj.cable_tree
 
-    assert len(diag.nodes) == elements_n
+    assert len(diag.nodes) >= number_of_elements