Revert functional test for comprehension ==> this now works with new util functions

Author: Julfried

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

@@ -1,4 +1,10 @@
 classDiagram
+  class AggregationContainer {
+    component_dict : dict[str, Component]
+    components : list[Component]
+    components_set : set[Component]
+    lazy_components : Generator[Component]
+  }
   class AssociationContainer {
     component_dict : dict[int, Component]
     components : list[Component]
@@ -22,3 +28,7 @@ classDiagram
   Component --* CompositionContainer : component_dict
   Component --* CompositionContainer : components_set
   Component --* CompositionContainer : lazy_components
+  Component --o AggregationContainer : components
+  Component --o AggregationContainer : component_dict
+  Component --o AggregationContainer : components_set
+  Component --o AggregationContainer : lazy_components

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

@@ -17,6 +17,15 @@ def __init__(self):
         self.components_set: set[Component]
         self.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):