Update the functional test for comprehensions aswell

Author: Julfried

tests/pyreverse/functional/class_diagrams/associations/comprehensions.mmd

@@ -2,13 +2,33 @@ classDiagram
   class Component {
     name : str
   }
-  class Container {
+  class AssociationContainer {
+    components : list[Component]
     component_dict : dict[int, Component]
+    components_set : set[Component]
+    lazy_components : Generator[Component]
+  }
+  class AggregationContainer {
     components : list[Component]
+    component_dict : dict[str, Component]
+    components_set : set[Component]
+    lazy_components : Generator[Component]
+  }
+  class CompositionContainer {
+    components : list[Component]
+    component_dict : dict[int, Component]
     components_set : set[Component]
     lazy_components : Generator[Component]
   }
-  Component --o Container : components
-  Component --o Container : component_dict
-  Component --o Container : components_set
-  Component --o Container : lazy_components
+  Component --> AssociationContainer : components
+  Component --> AssociationContainer : component_dict
+  Component --> AssociationContainer : components_set
+  Component --> AssociationContainer : lazy_components
+  Component --o AggregationContainer : components
+  Component --o AggregationContainer : component_dict
+  Component --o AggregationContainer : components_set
+  Component --o AggregationContainer : lazy_components
+  Component --* CompositionContainer : components
+  Component --* CompositionContainer : component_dict
+  Component --* CompositionContainer : components_set
+  Component --* CompositionContainer : lazy_components

tests/pyreverse/functional/class_diagrams/associations/comprehensions.py

@@ -8,10 +8,30 @@ def __init__(self, name: str):
         self.name = name
 
 
-class Container:
-    """A container class that uses comprehension to create components."""
+class AssociationContainer:
+    """Type hints only - no ownership."""
+    # Association: just type hints, no actual assignment
+    components: list[Component]
+    component_dict: dict[int, Component]
+    components_set: set[Component]
+    lazy_components: Generator[Component]
+
+
+class AggregationContainer:
+    """Comprehensions using existing objects - aggregation."""
+    def __init__(self, existing_components: list[Component]):
+        # Aggregation: comprehensions using existing objects (not creating)
+        self.components: list[Component] = [comp for comp in existing_components]
+        self.component_dict: dict[str, Component] = {f"key_{i}": comp for i, comp in enumerate(existing_components)}
+        self.components_set: set[Component] = {comp for comp in existing_components}
+        self.lazy_components: Generator[Component] = (comp for comp in existing_components)
+
+
+class CompositionContainer:
+    """Comprehensions creating new objects - composition."""
     def __init__(self):
-        self.components: list[Component] = [Component(f"component_{i}") for i in range(3)] # list
-        self.component_dict: dict[int, Component] = {i: Component(f"dict_component_{i}") for i in range(2)} # dict
-        self.components_set: set[Component] = {Component(f"set_component_{i}") for i in range(2)} # set
-        self.lazy_components: Generator[Component] = (Component(f"lazy_{i}") for i in range(2)) # generator
+        # Composition: comprehensions creating new objects
+        self.components: list[Component] = [Component(f"component_{i}") for i in range(3)]
+        self.component_dict: dict[int, Component] = {i: Component(f"dict_component_{i}") for i in range(2)}
+        self.components_set: set[Component] = {Component(f"set_component_{i}") for i in range(2)}
+        self.lazy_components: Generator[Component] = (Component(f"lazy_{i}") for i in range(2))