wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-06-28 23:57:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added test case for grid_shape parameter.

Browse Source
This commit is contained in:
Horea Christian
2013-07-21 03:55:16 +02:00
parent 228bf8f7ba
commit 2d33bf7622
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/util/tests/test_montage.py
+17
View File
@@ -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():