From 2d33bf7622ca1f5ed5cebdfaf7f27919ffbf850d Mon Sep 17 00:00:00 2001
From: Horea Christian <h.chr@mail.ru>
Date: Sun, 21 Jul 2013 03:55:16 +0200
Subject: [PATCH] Added test case for grid_shape parameter.

---
 skimage/util/tests/test_montage.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/skimage/util/tests/test_montage.py b/skimage/util/tests/test_montage.py
index 47e5426d..9de3f3cd 100644
--- a/skimage/util/tests/test_montage.py
+++ b/skimage/util/tests/test_montage.py
@@ -51,6 +51,23 @@ def test_shape():
 
     arr_out = montage2d(arr_in)
     assert_equal(arr_out.shape, (alpha * height, alpha * width))
+    
+    
+def test_grid_shape():
+    n_images = 6
+    height, width = 2, 2
+    arr_in = np.arange(n_images * height * width, dtype=np.float32)
+    arr_in = arr_in.reshape(n_images, height, width)
+    arr_out = montage.montage2d(arr_in, grid_shape=(3,2))
+    correct_arr_out = np.array(
+	[[  0.,   1.,   4.,   5.],
+	 [  2.,   3.,   6.,   7.],
+	 [  8.,   9.,  12.,  13.],
+	 [ 10.,  11.,  14.,  15.],
+	 [ 16.,  17.,  20.,  21.],
+	 [ 18.,  19.,  22.,  23.]]
+    )
+    assert_array_equal(arr_out, correct_arr_out)
 
 
 def test_rescale_intensity():