From d0e7a26f6fcb19013e018cf11f1137ffb0a19b7d Mon Sep 17 00:00:00 2001 From: Vighnesh Birodkar Date: Thu, 19 Jun 2014 16:36:17 +0530 Subject: [PATCH] Added RAG merge_nodes test case --- skimage/graph/tests/test_rag.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/skimage/graph/tests/test_rag.py b/skimage/graph/tests/test_rag.py index e239e79f..39e99852 100644 --- a/skimage/graph/tests/test_rag.py +++ b/skimage/graph/tests/test_rag.py @@ -1,5 +1,22 @@ import numpy as np from skimage import graph +import random + + +def test_rag_merge(): + g = graph.rag.RAG() + for i in range(10): + g.add_edge(i, (i + 1) % 10, {'weight': i * 10}) + g.node[i]['labels'] = [i] + + for i in range(9): + x = random.choice(g.nodes()) + y = random.choice(g.neighbors(x)) + g.merge_nodes(x, y) + + idx = g.nodes()[0] + sorted(g.node[idx]['labels']) == range(10) + assert g.edges() == [] def test_threshold_cut():