mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 10:12:05 +08:00
Stefan van der Walt
18 lines
518 B
Python
18 lines
518 B
Python
import numpy as np
|
|
from numpy.testing import assert_array_almost_equal
|
|
|
|
from skimage import transform as tf, data, img_as_float
|
|
|
|
|
|
def test_roundtrip():
|
|
image = img_as_float(data.checkerboard())
|
|
|
|
swirl_params = {'radius': 80, 'rotation': 0, 'order': 2, 'mode': 'reflect'}
|
|
swirled = tf.swirl(image, strength=10, **swirl_params)
|
|
unswirled = tf.swirl(swirled, strength=-10, **swirl_params)
|
|
|
|
assert np.mean(np.abs(image - unswirled)) < 0.01
|
|
|
|
if __name__ == "__main__":
|
|
np.testing.run_module_suite()
|