add edge method

txiong000 · Feb 4, 2019 · 2a06c29 · 2a06c29
1 parent 99b8b9a
commit 2a06c29
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/projects/graph/backup/graph.py b/projects/graph/backup/graph.py
@@ -12,6 +12,8 @@ def __init__(self):
     def add_vertex(self, vertex):
         self.vertices[vertex] = set()
         pass
-    def add_edge(self):
+    def add_edge(self, vert_1, vert_2):
+        self.vertices[vert_1].add(vert_2)
+        self.vertices[vert_2].add(vert_1)
         pass