Test add_edge in CheckCycleUndirectedGraph

Tjstretchalot · Tjstretchalot · commit 47d31876e15a · 2017-08-19T09:22:17.000-07:00
This test is failing and is showing why the current
implementation for CheckCycleUndirectedGraph is not
correct.
diff --git a/tests/test_data_structure.py b/tests/test_data_structure.py
@@ -192,6 +192,24 @@ def test_cycle_in_directed_graph(self):
         myGraph.add_edge(3, 3)
 
         self.assertTrue(myGraph.check_cycle())
+    
+    def test_add_edge_in_undirected_graph(self):
+        myGraph = graph.CheckCycleUndirectedGraph()
+        myGraph.add_edge(0, 1)
+        myGraph.add_edge(0, 2)
+        
+        setFrom0 = myGraph.graph[0]
+        setFrom1 = myGraph.graph[1]
+        setFrom2 = myGraph.graph[2]
+        
+        self.assertIsNotNone(setFrom0)
+        self.assertIsNotNone(setFrom1)
+        self.assertIsNotNone(setFrom2)
+        
+        self.assertIn(1, setFrom0)
+        self.assertIn(0, setFrom1)
+        self.assertIn(2, setFrom0)
+        self.assertIn(0, setFrom2)
 
     def test_cycle_in_undirected_graph(self):
         myGraph = graph.CheckCycleUndirectedGraph()
