fix: Apply suggestions from code review

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

Hide file tree

Showing 2 changed files with 2 additions and 3 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
@@ -22,9 +22,9 @@
 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