mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 04:55:16 +08:00
Rebecca
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from numpy.testing import assert_array_equal, assert_raises, run_module_suite
|
|
from skimage import data
|
|
from skimage.transform import pyramids
|
|
|
|
|
|
image = data.astronaut()
|
|
image_gray = image[..., 0]
|
|
|
|
|
|
def test_pyramid_reduce_rgb():
|
|
rows, cols, dim = image.shape
|
|
out = pyramids.pyramid_reduce(image, downscale=2)
|
|
assert_array_equal(out.shape, (rows / 2, cols / 2, dim))
|
|
|
|
|
|
def test_pyramid_reduce_gray():
|
|
rows, cols = image_gray.shape
|
|
out = pyramids.pyramid_reduce(image_gray, downscale=2)
|
|
assert_array_equal(out.shape, (rows / 2, cols / 2))
|
|
|
|
|
|
def test_pyramid_expand_rgb():
|
|
rows, cols, dim = image.shape
|
|
out = pyramids.pyramid_expand(image, upscale=2)
|
|
assert_array_equal(out.shape, (rows * 2, cols * 2, dim))
|
|
|
|
|
|
def test_pyramid_expand_gray():
|
|
rows, cols = image_gray.shape
|
|
out = pyramids.pyramid_expand(image_gray, upscale=2)
|
|
assert_array_equal(out.shape, (rows * 2, cols * 2))
|
|
|
|
|
|
def test_build_gaussian_pyramid_rgb():
|
|
rows, cols, dim = image.shape
|
|
pyramid = pyramids.pyramid_gaussian(image, downscale=2)
|
|
for layer, out in enumerate(pyramid):
|
|
layer_shape = (rows / 2 ** layer, cols / 2 ** layer, dim)
|
|
assert_array_equal(out.shape, layer_shape)
|
|
|
|
|
|
def test_build_gaussian_pyramid_gray():
|
|
rows, cols = image_gray.shape
|
|
pyramid = pyramids.pyramid_gaussian(image_gray, downscale=2)
|
|
for layer, out in enumerate(pyramid):
|
|
layer_shape = (rows / 2 ** layer, cols / 2 ** layer)
|
|
assert_array_equal(out.shape, layer_shape)
|
|
|
|
|
|
def test_build_laplacian_pyramid_rgb():
|
|
rows, cols, dim = image.shape
|
|
pyramid = pyramids.pyramid_laplacian(image, downscale=2)
|
|
for layer, out in enumerate(pyramid):
|
|
layer_shape = (rows / 2 ** layer, cols / 2 ** layer, dim)
|
|
assert_array_equal(out.shape, layer_shape)
|
|
|
|
|
|
def test_build_laplacian_pyramid_gray():
|
|
rows, cols = image_gray.shape
|
|
pyramid = pyramids.pyramid_laplacian(image_gray, downscale=2)
|
|
for layer, out in enumerate(pyramid):
|
|
layer_shape = (rows / 2 ** layer, cols / 2 ** layer)
|
|
assert_array_equal(out.shape, layer_shape)
|
|
|
|
|
|
def test_check_factor():
|
|
assert_raises(ValueError, pyramids._check_factor, 0.99)
|
|
assert_raises(ValueError, pyramids._check_factor, - 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|