fix: Fix-class-tree (#122)

DSD-DBS · Jul 11, 2024 · fd9ad24 · fd9ad24
1 parent 5c22272
commit fd9ad24
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
@@ -353,7 +353,9 @@ def _create_diagram(self, params: dict[str, t.Any]) -> cdiagram.Diagram:
                 setattr(self, param_name, override)
 
         data = params.get("elkdata") or get_elkdata(self, params)
-        if has_single_child(data):
+        if not isinstance(
+            self, (ClassTreeDiagram, InterfaceContextDiagram)
+        ) and has_single_child(data):
             self.display_derived_interfaces = True
             data = get_elkdata(self, params)
 

diff --git a/tests/test_interface_diagrams.py b/tests/test_interface_diagrams.py
@@ -51,9 +51,6 @@ def test_interface_diagram_with_hide_functions(
     obj = model.by_uuid(TEST_INTERFACE_UUID)
 
     diag = obj.context_diagram.render(None, hide_functions=True)
-    obj.context_diagram.render("svgdiagram", hide_functions=True).save(
-        pretty=True
-    )
 
     for uuid in (
         "fbfb2b20-b711-4211-9b75-25e38390cdbc",  # LogicalFunction