From 61084612e76861b1acd87dd2860cd745025d737c Mon Sep 17 00:00:00 2001
From: Zachary Pincus <zpincus@gmail.com>
Date: Fri, 7 Jan 2011 08:58:05 -0500
Subject: [PATCH] add tests

---
 scikits/image/graph/tests/test_mcp.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/scikits/image/graph/tests/test_mcp.py b/scikits/image/graph/tests/test_mcp.py
index f9f62e8e..cb285618 100644
--- a/scikits/image/graph/tests/test_mcp.py
+++ b/scikits/image/graph/tests/test_mcp.py
@@ -43,6 +43,32 @@ def test_basic():
                         (6, 1),
                         (7, 2)])
 
+def test_neg_inf():
+    expected_costs = numpy.where(a==1, np.inf, 0)
+    expected_path = [(1, 6),
+                     (1, 5),
+                     (1, 4),
+                     (1, 3),
+                     (1, 2),
+                     (2, 1),
+                     (3, 1),
+                     (4, 1),
+                     (5, 1),
+                     (6, 1)]
+    test_neg = numpy.where(a==1, -1, 0)
+    test_inf = numpy.where(a==1, np.inf, 0)
+    m = mcp.MCP(test_neg, fully_connected=True)
+    costs, traceback = m.find_costs([(1, 6)])
+    return_path = m.traceback((6, 1))
+    assert_array_equal(costs, expected_costs)
+    assert_array_equal(return_path, expected_path)
+    m = mcp.MCP(test_inf, fully_connected=True)
+    costs, traceback = m.find_costs([(1, 6)])
+    return_path = m.traceback((6, 1))
+    assert_array_equal(costs, expected_costs)
+    assert_array_equal(return_path, expected_path)
+  
+
 def test_route():
     return_path, cost = mcp.route_through_array(a, (1,6), (7,2), geometric=True)
     assert_almost_equal(cost, np.sqrt(2)/2)