1 files changed, 46 insertions, 0 deletions
diff --git a/test/test_dag.rb b/test/test_dag.rb
new file mode 100755
index 0000000..f5330da
--- /dev/null
+++ b/test/test_dag.rb
@@ -0,0 +1,46 @@
+#!/usr/bin/env ruby
+
+require 'nlp_ruby'
+require 'test/unit'
+
+
+class TestDAG <  Test::Unit::TestCase
+
+  def test_viterbi
+    semiring = ViterbiSemiring.new
+    graph, nodes_by_label = DAG::read_graph_from_json('test/dag/example.json', semiring)
+    DAG::viterbi(graph, semiring, nodes_by_label['0'])
+    assert_equal(nodes_by_label['100'].score, 0.003)
+  end
+
+  def test_dijkstra
+    semiring = RealSemiring.new
+    graph, nodes_by_label = DAG::read_graph_from_json('test/dag/example.json', semiring)
+    DAG::dijkstra(graph, semiring, nodes_by_label['0'])
+    assert_equal(nodes_by_label['100'].score, 0.5)
+  end
+
+  def test_bellman_ford
+    semiring = RealSemiring.new
+    graph, nodes_by_label = DAG::read_graph_from_json('test/dag/example.json', semiring)
+    DAG::bellman_ford(graph, semiring, nodes_by_label['0'])
+    assert_equal(nodes_by_label['100'].score, 0.5)
+  end
+
+  def test_floyd
+    graph, _ = DAG::read_graph_from_json('test/dag/example.json')
+    d = DAG::floyd(graph)
+    assert_equal(d[0][graph.size-1], 0.5)
+  end
+
+  def test_dfs
+    _, nodes_by_label = DAG::read_graph_from_json('test/dag/example.json')
+    assert_equal(nodes_by_label['100'], DAG::dfs(nodes_by_label['0'], '100'))
+  end
+
+  def test_bfs
+    _, nodes_by_label = DAG::read_graph_from_json('test/dag/example.json')
+    assert_equal(nodes_by_label['100'], DAG::bfs(nodes_by_label['0'], '100'))
+  end
+end
+