update

Author: rain1024

visualize/graph.py

@@ -1,5 +1,3 @@
-import svgwrite
-
 class Graph:
     def __init__(self, directed=False):
         self.nodes = set()
@@ -21,37 +19,15 @@ def add_edge(self, u, v):
                 self.edges[v] = [u]
 
 
-
 def main():
     # Create a list of edges for a tree with 7 nodes
-    edges = {
-        1: [2, 3],
-        2: [3, 4, 5],
-        3: [6, 7]
-    }
+    edges = {1: [2, 3], 2: [3, 4, 5], 3: [6, 7]}
 
-if __name__ == '__main__':
     # Create a list of edges for a tree with 7 nodes
-    edges = {
-        1: [2, 3],
-        2: [3, 4, 5],
-        3: [6, 7]
-    }
+    edges = {1: [2, 3], 2: [3, 4, 5], 3: [6, 7]}
 
-    # Create an instance of the Graph class with 7 nodes and the list of edges
     graph = Graph(7, edges)
 
-    # Generate an SVG image of the tree and save it to a file
-    graph.renderSvg('graph.svg')
-
-class Visualizer:
-    def __init__(self):
-        pass
 
-    def visualize(self, graph):
-        # calculate the position of each nodes
-        positions = []
-        for i, u in enumerate(graph.nodes):
-            x = i * 30
-            y = 50
-            self.position.push((x, y))
+if __name__ == "__main__":
+    main()

visualize/tests/test_visualizer.py

@@ -0,0 +1,31 @@
+import unittest
+from typing import List
+
+from graphics import Circle, Rectangle, Line
+from graph import Graph
+from graphics import GraphicObject
+from visualizer import Visualizer
+
+class TestVisualizer(unittest.TestCase):
+    def test_visualize(self):
+        # Create a graph with two nodes and one edge
+        graph = Graph()
+        graph.add_edge(1, 2)
+
+        # Create an instance of Visualizer
+        visualizer = Visualizer()
+
+        # Call the visualize method and store the result in a variable
+        graphic_objects = visualizer.visualize(graph)
+
+        self.assertEqual(len(graphic_objects), 4)
+
+        # Check that the first element is a Circle
+        self.assertIsInstance(graphic_objects[0], Circle)
+
+        # Check that the second element is a Line
+        self.assertIsInstance(graphic_objects[3], Line)
+
+
+if __name__ == '__main__':
+    unittest.main()

visualize/visualizer.py

@@ -0,0 +1,29 @@
+from typing import List
+from graph import Graph
+from graphics import GraphicObject, Circle, Line
+
+
+class Visualizer:
+    def __init__(self):
+        pass
+
+    def visualize(self, graph: Graph) -> List[GraphicObject]:
+        """Generates a list of GraphicObject instances based on the given graph.
+
+        Args:
+            graph: A Graph object.
+
+        Returns:
+            A list of GraphicObject instances.
+        """
+        graphic_objects = []
+        positions = {}
+        for i, node in enumerate(graph.nodes):
+            x = i * 30
+            y = 50
+            positions[node] = (x, y)
+            graphic_objects.append(Circle((x, y), 10))
+        for u, vs in graph.edges.items():
+            for v in vs:
+                graphic_objects.append(Line(positions[u], positions[v]))
+        return graphic_objects