test: update findAllcycles tests

TheAlgorithms · Oct 25, 2023 · 4163638 · 4163638
1 parent fb7e6dd
commit 4163638
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/graph/cycle_test.go b/graph/cycle_test.go
@@ -37,4 +37,17 @@ func TestFindAllCycles(t *testing.T) {
 		t.Error("number of cycles is not correct")
 	}
 
+	firstCycle := res[0]
+	if len(firstCycle.edges) != 3 {
+		t.Error("number of vertex in cycle is not correct")
+	}
+	if _, ok := firstCycle.edges[3][4]; !ok {
+		t.Error("connection in cycle is not correct")
+	}
+	if _, ok := firstCycle.edges[4][7]; !ok {
+		t.Error("connection in cycle is not correct")
+	}
+	if _, ok := firstCycle.edges[7][3]; !ok {
+		t.Error("connection in cycle is not correct")
+	}
 }