update

Author: rain1024

.gitignore

@@ -7,3 +7,6 @@ bazel-testlogs
 bazel-*
 *.md2
 test_data
+__pycache__
+htmlcov
+.coverage

visualize/.coveragerc

@@ -0,0 +1,4 @@
+[report]
+exclude_lines =
+    unittest.main()
+    main()

visualize/__init__.py

@@ -0,0 +1,64 @@
+class Color:
+    RED_500 = '#F44336'
+    RED_50 = '#FFEBEE'
+    RED_100 = '#FFCDD2'
+    RED_200 = '#EF9A9A'
+    RED_300 = '#E57373'
+    RED_400 = '#EF5350'
+    RED_500 = '#F44336'
+    RED_600 = '#E53935'
+    RED_700 = '#D32F2F'
+    RED_800 = '#C62828'
+    RED_900 = '#B71C1C'
+    RED_A100 = '#FF8A80'
+    RED_A200 = '#FF5252'
+    RED_A400 = '#FF1744'
+    RED_A700 = '#D50000'
+
+    PINK_500 = '#E91E63'
+    PINK_50 = '#FCE4EC'
+    PINK_100 = '#F8BBD0'
+    PINK_200 = '#F48FB1'
+    PINK_300 = '#F06292'
+    PINK_400 = '#EC407A'
+    PINK_500 = '#E91E63'
+    PINK_600 = '#D81B60'
+    PINK_700 = '#C2185B'
+    PINK_800 = '#AD1457'
+    PINK_900 = '#880E4F'
+    PINK_A100 = '#FF80AB'
+    PINK_A200 = '#FF4081'
+    PINK_A400 = '#F50057'
+    PINK_A700 = '#C51162'
+
+    PURPLE_500 = '#9C27B0'
+    PURPLE_50 = '#F3E5F5'
+    PURPLE_100 = '#E1BEE7'
+    PURPLE_200 = '#CE93D8'
+    PURPLE_300 = '#BA68C8'
+    PURPLE_400 = '#AB47BC'
+    PURPLE_500 = '#9C27B0'
+    PURPLE_600 = '#8E24AA'
+    PURPLE_700 = '#7B1FA2'
+    PURPLE_800 = '#6A1B9A'
+    PURPLE_900 = '#4A148C'
+    PURPLE_A100 = '#EA80FC'
+    PURPLE_A200 = '#E040FB'
+    PURPLE_A400 = '#D500F9'
+    PURPLE_A700 = '#AA00FF'
+
+    BLUE_500 = '#2196F3'
+    BLUE_50 = '#E3F2FD'
+    BLUE_100 = '#BBDEFB'
+    BLUE_200 = '#90CAF9'
+    BLUE_300 = '#64B5F6'
+    BLUE_400 = '#42A5F5'
+    BLUE_500 = '#2196F3'
+    BLUE_600 = '#1E88E5'
+    BLUE_700 = '#1976D2'
+    BLUE_800 = '#1565C0'
+    BLUE_900 = '#0D47A1'
+    BLUE_A100 = '#82B1FF'
+    BLUE_A200 = '#448AFF'
+    BLUE_A400 = '#2979FF'
+    BLUE_A700 = '#2962FF'

visualize/colors.py

@@ -0,0 +1,57 @@
+import svgwrite
+
+class Graph:
+    def __init__(self, directed=False):
+        self.nodes = set()
+        self.edges = {}
+        self.directed = directed
+
+    def add_edge(self, u, v):
+        self.nodes.add(u)
+        self.nodes.add(v)
+        if u in self.edges:
+            self.edges[u].append(v)
+        else:
+            self.edges[u] = [v]
+        if not self.directed:
+            if v in self.edges:
+                self.edges[v].append(u)
+            else:
+                self.edges[v] = [u]
+                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]
+    }
+
+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]
+    }
+
+    # 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))

visualize/graph.py

@@ -0,0 +1,22 @@
+class GraphicObject:
+    def __init__(self, type, **kwargs):
+        self.type = type
+        self.kwargs = kwargs
+
+class Circle(GraphicObject):
+    def __init__(self, center, radius, **kwargs):
+        super().__init__("circle", **kwargs)
+        self.center = center
+        self.radius = radius
+
+class Rectangle(GraphicObject):
+    def __init__(self, top_left, bottom_right, **kwargs):
+        super().__init__("rectangle", **kwargs)
+        self.top_left = top_left
+        self.bottom_right = bottom_right
+
+class Line(GraphicObject):
+    def __init__(self, start, end, **kwargs):
+        super().__init__("line", **kwargs)
+        self.start = start
+        self.end = end

visualize/graph.svg

@@ -0,0 +1,49 @@
+import unittest
+
+from graph import Graph
+
+class TestGraph(unittest.TestCase):
+    def test_init(self):
+        graph = Graph()
+        self.assertEqual(graph.nodes, set())
+        self.assertEqual(graph.edges, {})
+        self.assertFalse(graph.directed)
+
+        directed_graph = Graph(directed=True)
+        self.assertTrue(directed_graph.directed)
+
+    def test_add_edge(self):
+        graph = Graph()
+        graph.add_edge(1, 2)
+        self.assertEqual(graph.nodes, {1, 2})
+        self.assertEqual(graph.edges, {1: [2], 2: [1]})
+
+        graph.add_edge(2, 3)
+        self.assertEqual(graph.nodes, {1, 2, 3})
+        self.assertEqual(graph.edges, {1: [2], 2: [1, 3], 3: [2]})
+
+        directed_graph = Graph(directed=True)
+        directed_graph.add_edge(1, 2)
+        self.assertEqual(directed_graph.nodes, {1, 2})
+        self.assertEqual(directed_graph.edges, {1: [2]})
+    
+    def test_add_edge_undirected(self):
+        graph = Graph(directed=False)
+        graph.add_edge(1, 2)
+        self.assertEqual(graph.nodes, {1, 2})
+        self.assertEqual(graph.edges, {1: [2], 2: [1]})
+
+        graph.add_edge(2, 3)
+        self.assertEqual(graph.nodes, {1, 2, 3})
+        self.assertEqual(graph.edges, {1: [2], 2: [1, 3], 3: [2]})
+    
+    def test_add_edge_to_existing_node(self):
+        graph = Graph(directed=False)
+        graph.add_edge(1, 2)
+        graph.add_edge(1, 3)
+        graph.add_edge(2, 3)
+        self.assertEqual(graph.nodes, {1, 2, 3})
+        self.assertEqual(graph.edges, {1: [2, 3], 2: [1, 3], 3: [1, 2]})
+
+if __name__ == '__main__':
+    unittest.main()

visualize/graphics.py

@@ -0,0 +1,82 @@
+import unittest
+
+from graphics import Circle, Rectangle, Line
+
+class TestCircle(unittest.TestCase):
+    def test_init(self):
+        circle = Circle((0, 0), 3, color='red', filled=True)
+        self.assertEqual(circle.type, 'circle')
+        self.assertEqual(circle.center, (0, 0))
+        self.assertEqual(circle.radius, 3)
+        self.assertEqual(circle.kwargs, {'color': 'red', 'filled': True})
+
+    def test_center_as_float(self):
+        circle = Circle((0.5, 0.5), 3, color='red', filled=True)
+        self.assertEqual(circle.center, (0.5, 0.5))
+
+    def test_negative_radius(self):
+        circle = Circle((0, 0), -3, color='red', filled=True)
+        self.assertEqual(circle.radius, -3)
+
+    def test_color_as_none(self):
+        circle = Circle((0, 0), 3, color=None, filled=True)
+        self.assertEqual(circle.kwargs, {'color': None, 'filled': True})
+
+    def test_additional_kwargs(self):
+        circle = Circle((0, 0), 3, color='red', filled=True, linewidth=2)
+        self.assertEqual(circle.kwargs, {'color': 'red', 'filled': True, 'linewidth': 2})
+
+class TestRectangle(unittest.TestCase):
+    def test_init(self):
+        rectangle = Rectangle((0, 0), (3, 4), color='blue', filled=False)
+        self.assertEqual(rectangle.type, 'rectangle')
+        self.assertEqual(rectangle.top_left, (0, 0))
+        self.assertEqual(rectangle.bottom_right, (3, 4))
+        self.assertEqual(rectangle.kwargs, {'color': 'blue', 'filled': False})
+
+    def test_coordinates_as_float(self):
+        rectangle = Rectangle((0.5, 0.5), (3.5, 4.5), color='blue', filled=False)
+        self.assertEqual(rectangle.top_left, (0.5, 0.5))
+        self.assertEqual(rectangle.bottom_right, (3.5, 4.5))
+
+    def test_negative_coordinates(self):
+        rectangle = Rectangle((-1, -1), (-2, -2), color='blue', filled=False)
+        self.assertEqual(rectangle.top_left, (-1, -1))
+        self.assertEqual(rectangle.bottom_right, (-2, -2))
+
+    def test_color_as_none(self):
+        rectangle = Rectangle((0, 0), (3, 4), color=None, filled=False)
+        self.assertEqual(rectangle.kwargs, {'color': None, 'filled': False})
+
+    def test_additional_kwargs(self):
+        rectangle = Rectangle((0, 0), (3, 4), color='blue', filled=False, linewidth=2)
+        self.assertEqual(rectangle.kwargs, {'color': 'blue', 'filled': False, 'linewidth': 2})
+
+class TestLine(unittest.TestCase):
+    def test_init(self):
+        line = Line((0, 0), (3, 4), color='green', dashed=True)
+        self.assertEqual(line.type, 'line')
+        self.assertEqual(line.start, (0, 0))
+        self.assertEqual(line.end, (3, 4))
+        self.assertEqual(line.kwargs, {'color': 'green', 'dashed': True})
+
+    def test_coordinates_as_float(self):
+        line = Line((0.5, 0.5), (3.5, 4.5), color='green', dashed=True)
+        self.assertEqual(line.start, (0.5, 0.5))
+        self.assertEqual(line.end, (3.5, 4.5))
+
+    def test_negative_coordinates(self):
+        line = Line((-1, -1), (-2, -2), color='green', dashed=True)
+        self.assertEqual(line.start, (-1, -1))
+        self.assertEqual(line.end, (-2, -2))
+
+    def test_color_as_none(self):
+        line = Line((0, 0), (3, 4), color=None, dashed=True)
+        self.assertEqual(line.kwargs, {'color': None, 'dashed': True})
+
+    def test_additional_kwargs(self):
+        line = Line((0, 0), (3, 4), color='green', dashed=True, linewidth=2)
+        self.assertEqual(line.kwargs, {'color': 'green', 'dashed': True, 'linewidth': 2})
+        
+if __name__ == '__main__':
+    unittest.main()