update

Author: rain1024

visualize/graph.py

@@ -26,7 +26,7 @@ def main():
     # Create a list of edges for a tree with 7 nodes
     edges = {1: [2, 3], 2: [3, 4, 5], 3: [6, 7]}
 
-    graph = Graph(7, edges)
+    graph = Graph()
 
 
 if __name__ == "__main__":

visualize/graph.svg

@@ -1,10 +1,11 @@
 import unittest
-from typing import List
-
 from graphics import Circle, Rectangle, Line
 from graph import Graph
 from graphics import GraphicObject
-from visualizer import Visualizer
+from visualizer import Visualizer, SvgVisualizer
+import svgwrite
+import os
+
 
 class TestVisualizer(unittest.TestCase):
     def test_visualize(self):
@@ -27,5 +28,57 @@ def test_visualize(self):
         self.assertIsInstance(graphic_objects[3], Line)
 
 
-if __name__ == '__main__':
+class TestSvgVisualizer(unittest.TestCase):
+    def setUp(self):
+        self.skip_tear_down = False
+        self.filepath = ""
+
+    def tearDown(self):
+        if self.skip_tear_down:
+            return
+        # Delete the file created by the test
+        try:
+            os.remove(self.filepath)
+        except FileNotFoundError:
+            pass
+
+    def test_draw(self):
+        # Create a Graph object with a few nodes and edges
+        graph = Graph()
+        graph.add_edge(1, 2)
+        graph.add_edge(2, 3)
+        graph.add_edge(3, 4)
+
+        # Create an instance of SvgVisualizer
+        visualizer = SvgVisualizer()
+
+        # Call the draw method
+        objects = visualizer.visualize(graph)
+        drawing = visualizer.draw(objects)
+
+        # Check that the returned value is an svgwrite.Drawing object
+        self.assertIsInstance(drawing, svgwrite.Drawing)
+
+    def test_write(self):
+        # Create a Graph object with a few nodes and edges
+        self.filepath = "test.svg"
+        graph = Graph()
+        graph.add_edge(1, 2)
+        graph.add_edge(2, 3)
+        graph.add_edge(3, 4)
+
+        # Create an instance of SvgVisualizer
+        visualizer = SvgVisualizer()
+        graphic_objects = visualizer.visualize(graph)
+
+        # Call the write method
+        visualizer.write(self.filepath, graphic_objects)
+
+        # Check that the file was created
+        self.assertTrue(os.path.exists(self.filepath))
+
+        self.skip_tear_down = True
+
+
+if __name__ == "__main__":
     unittest.main()

visualize/image.svg

@@ -1,6 +1,7 @@
 from typing import List
 from graph import Graph
 from graphics import GraphicObject, Circle, Line
+import svgwrite
 
 
 class Visualizer:
@@ -27,3 +28,62 @@ def visualize(self, graph: Graph) -> List[GraphicObject]:
             for v in vs:
                 graphic_objects.append(Line(positions[u], positions[v]))
         return graphic_objects
+
+
+class SvgVisualizer(Visualizer):
+    def __init__(self):
+        super().__init__()
+
+    def draw(self, graphic_objects: List[GraphicObject]) -> svgwrite.Drawing:
+        """Converts a list of GraphicObject instances to an SVG Drawing object.
+
+        Args:
+            graphic_objects: A list of GraphicObject instances.
+
+        Returns:
+            An svgwrite.Drawing object.
+        """
+        dwg = svgwrite.Drawing()
+        for obj in graphic_objects:
+            if obj.type == "circle":
+                dwg.add(self._circle_to_svg(obj))
+            elif obj.type == "line":
+                dwg.add(self._line_to_svg(obj))
+            # Add code to handle other types of GraphicObjects
+        return dwg
+
+    def write(self, filepath: str, graphic_objects: List[GraphicObject]):
+        """Writes a list of GraphicObject instances to an SVG file.
+
+        Args:
+            filepath: The path to the output file.
+            graphic_objects: A list of GraphicObject instances.
+        """
+        dwg = self.draw(graphic_objects)
+        dwg.saveas(filepath)
+
+    def _circle_to_svg(self, circle: Circle) -> svgwrite.shapes.Circle:
+        """Converts a Circle instance to an SVG Circle object.
+
+        Args:
+            circle: A Circle instance.
+
+        Returns:
+            An svgwrite.shapes.Circle object.
+        """
+        x, y = circle.center
+        r = circle.radius
+        return svgwrite.shapes.Circle(center=(x, y), r=r)
+
+    def _line_to_svg(self, line: Line) -> svgwrite.shapes.Line:
+        """Converts a Line instance to an SVG Line object.
+
+        Args:
+            line: A Line instance.
+
+        Returns:
+            An svgwrite.shapes.Line object.
+        """
+        x1, y1 = line.start
+        x2, y2 = line.end
+        return svgwrite.shapes.Line(start=(x1, y1), end=(x2, y2))