From f9dfa1d81d139dc833af5870c7c8b1e1604ffc25 Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Fri, 8 Jan 2016 17:25:10 +1100 Subject: [PATCH] Make mean boundary RAG testing more robust --- skimage/future/graph/tests/test_rag.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/skimage/future/graph/tests/test_rag.py b/skimage/future/graph/tests/test_rag.py index dafa0ddc..27d31a99 100644 --- a/skimage/future/graph/tests/test_rag.py +++ b/skimage/future/graph/tests/test_rag.py @@ -211,5 +211,7 @@ def test_rag_boundary(): labels[8:, 8:] = 4 g = graph.rag_boundary(labels, edge_map, connectivity=1) - assert len(g.nodes()) == 4 - assert len(g.edges()) == 4 + assert set(g.nodes()) == set([1, 2, 3, 4]) + assert set(g.edges()) == set([(1, 2), (1, 3), (2, 4), (3, 4)]) + assert g[1][3]['weight'] == 0.25 + assert g[2][4]['weight'] == 0.34375