From ecdfd518c6aa9a5406f86220411aa92d8b99d478 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= <jschoenberger@demuc.de>
Date: Tue, 9 Apr 2013 22:54:57 +0200
Subject: [PATCH] Add test case for number of labels

---
 skimage/morphology/tests/test_ccomp.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/skimage/morphology/tests/test_ccomp.py b/skimage/morphology/tests/test_ccomp.py
index 1e0829ca..1169be85 100644
--- a/skimage/morphology/tests/test_ccomp.py
+++ b/skimage/morphology/tests/test_ccomp.py
@@ -82,6 +82,14 @@ class TestConnectedComponents:
                             [-1,  0, -1],
                             [-1, -1, -1]])
 
+    def test_return_num(self):
+        x = np.array([[1, 0, 6],
+                      [0, 0, 6],
+                      [5, 5, 5]])
+
+        assert_array_equal(label(x, return_num=True)[1], 4)
+        assert_array_equal(label(x, background=0, return_num=True)[1], 3)
+
 
 if __name__ == "__main__":
     run_module_suite()