From 50e155e254f9d6f6846aa803eca9c2721b38c01c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Mon, 9 Dec 2013 11:43:32 +0100 Subject: [PATCH] Fix test case --- skimage/feature/tests/test_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skimage/feature/tests/test_template.py b/skimage/feature/tests/test_template.py index c9052d82..68eb3ead 100644 --- a/skimage/feature/tests/test_template.py +++ b/skimage/feature/tests/test_template.py @@ -129,7 +129,7 @@ def test_3d(): result = match_template(image, template) assert_equal(result.shape, (10, 10, 10)) - assert_equal(np.unravel_index(result.argmax(), image.shape), (2, 5, 6)) + assert_equal(np.unravel_index(result.argmax(), result.shape), (3, 5, 4)) def test_3d_pad_input(): @@ -142,7 +142,7 @@ def test_3d_pad_input(): result = match_template(image, template, pad_input=True) assert_equal(result.shape, (12, 12, 12)) - assert_equal(np.unravel_index(result.argmax(), image.shape), (4, 6, 5)) + assert_equal(np.unravel_index(result.argmax(), result.shape), (4, 6, 5)) def test_wrong_input():