add graph topology test

marlboroKay · Jan 30, 2019 · efeef70 · efeef70
1 parent a5f9749
commit efeef70
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/scala/src/test/scala/ch43_topology_sort/GraphTopologyTest.scala b/scala/src/test/scala/ch43_topology_sort/GraphTopologyTest.scala
@@ -42,4 +42,23 @@ class GraphTopologyTest extends FlatSpec with Matchers {
     val seq = graphTopology.topologySortByKahn()
     seq.map(nodes(_)).mkString(",") should equal("a,b,d,e,c")
   }
+
+  /*
+     a -> d <- b
+          |   /|\
+         \|/   |
+          e -> c
+   */
+  it should "topologySortByKahn - 3" in {
+    val nodes = Array("a", "b", "c", "d", "e")
+    val graphTopology = new GraphTopology(nodes.length)
+    graphTopology.addEdge(0, 3)
+    graphTopology.addEdge(3, 4)
+    graphTopology.addEdge(4, 2)
+    graphTopology.addEdge(2, 1)
+    graphTopology.addEdge(1, 3)
+
+    val seq = graphTopology.topologySortByKahn()
+    seq.map(nodes(_)).mkString(",") should equal("a")
+  }
 }