From 27026cea8735cf352d0d5a3c5a68a253156369c4 Mon Sep 17 00:00:00 2001 From: Umesh Date: Mon, 22 Apr 2013 07:51:32 -0700 Subject: [PATCH] updated tests for Roberts Edge Detector --- skimage/filter/tests/test_edges.py | 34 +++++++++--------------------- 1 file changed, 10 insertions(+), 24 deletions(-) diff --git a/skimage/filter/tests/test_edges.py b/skimage/filter/tests/test_edges.py index fec01109..e84dba41 100644 --- a/skimage/filter/tests/test_edges.py +++ b/skimage/filter/tests/test_edges.py @@ -12,16 +12,11 @@ def test_roberts_zeros(): def test_roberts_diagonal1(): """Roberts' on an edge should be a diagonal""" - i = (10, 10) n = (10, 10) - i = np.ones(i) - i = np.triu(i) + i = np.tri(10,10,0) image = (i > 0).astype(float) - n = np.zeros(n) - for k in range(2, 10): - for t in range(0, k-1): - n[k][t] = 1 - n[t][k] = 1 + n = np.tri(10,10,-2) + n = n+n.transpose() j = np.identity(10) result = F.roberts(image) j[9][9] = j[0][0] = 0 @@ -30,27 +25,18 @@ def test_roberts_diagonal1(): def test_roberts_diagonal2(): """Roberts' on an edge should be a diagonal""" - i = (10, 10) - j = (10, 10) - n = (10, 10) - i = np.ones(i) - for k in range(0, 10): - for t in range(10-k, 10): - i[k][t] = 0 - n = np.zeros(n) - for k in range(0, 10): - for t in range(0, 9-k): - n[k][t] = 0 - n[t][k] = 0 + i = np.tri(10,10,0,dtype=int) + i = np.rot90(i.transpose()) + n = np.tri(10,10,-2,dtype=int) + n = np.rot90(n.transpose()) + np.rot90(n) image = (i > 0).astype(float) - j = np.ones(j) - for k in range(0, 10): - j[k][9-k] = 5 + j = np.identity(10) + j = np.rot90(j) result = F.roberts(image) j[0][9] = j[9][0] = 0 - assert (np.all(result[j == 5] == 1) and np.all(result[n == 1] == 0)) + assert (np.all(result[j == 1] == 1) and np.all(result[n == 1] == 0)) def test_sobel_zeros():