From 073f28e90e27e9503eb998f896d0581595629f1f Mon Sep 17 00:00:00 2001
From: Pratap Vardhan <pratapgr8@gmail.com>
Date: Sun, 14 Dec 2014 10:52:53 +0530
Subject: [PATCH] TST: Test for selem.cube

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

diff --git a/skimage/morphology/tests/test_selem.py b/skimage/morphology/tests/test_selem.py
index 7f4cc7c3..35559fe4 100644
--- a/skimage/morphology/tests/test_selem.py
+++ b/skimage/morphology/tests/test_selem.py
@@ -30,6 +30,13 @@ class TestSElem():
                 expected_mask = np.ones((i, j), dtype='uint8')
                 assert_equal(expected_mask, actual_mask)
 
+    def test_cube_selem(self):
+        """Test cube structuring elements"""
+        for k in range(0, 5):
+            actual_mask = selem.cube(k)
+            expected_mask = np.ones((k, k, k), dtype='uint8')
+            assert_equal(expected_mask, actual_mask)
+
     def strel_worker(self, fn, func):
         matlab_masks = np.load(os.path.join(data_dir, fn))
         k = 0