From e80dd18d7e7dbdfeaad63b384724d5b0fa816e4e Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Fri, 12 Jul 2013 18:26:11 +0200 Subject: [PATCH] Add tests for 'regular_grid()' --- skimage/util/tests/test_regular_grid.py | 33 +++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 skimage/util/tests/test_regular_grid.py diff --git a/skimage/util/tests/test_regular_grid.py b/skimage/util/tests/test_regular_grid.py new file mode 100644 index 00000000..1216b29d --- /dev/null +++ b/skimage/util/tests/test_regular_grid.py @@ -0,0 +1,33 @@ +import numpy as np +from nose.tools import raises +from numpy.testing import assert_equal +from skimage.util.regular_grid import regular_grid + + +def test_regular_grid_2d_8(): + ar = np.zeros((20, 40)) + g = regular_grid(ar.shape, 8) + assert_equal(g, [slice(5.0, None, 10.0), slice(5.0, None, 10.0)]) + ar[g] = 1 + assert_equal(ar.sum(), 8) + + +def test_regular_grid_2d_32(): + ar = np.zeros((20, 40)) + g = regular_grid(ar.shape, 32) + assert_equal(g, [slice(2.0, None, 5.0), slice(2.0, None, 5.0)]) + ar[g] = 1 + assert_equal(ar.sum(), 32) + + +def test_regular_grid_3d_8(): + ar = np.zeros((3, 20, 40)) + g = regular_grid(ar.shape, 8) + assert_equal(g, [slice(1.0, None, 3.0), slice(5.0, None, 10.0), + slice(5.0, None, 10.0)]) + ar[g] = 1 + assert_equal(ar.sum(), 8) + + +if __name__ == '__main__': + np.testing.run_module_suite()