STY: Cleanups after Tony's review.

This commit is contained in:
Stefan van der Walt
2012-05-04 12:16:05 -07:00
parent e13cc2541f
commit a5d8593408
4 changed files with 49 additions and 31 deletions
+3 -4
View File
@@ -6,11 +6,10 @@ 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'}
unswirled = tf.swirl(
tf.swirl(image, strength=10, **swirl_params),
strength=-10, **swirl_params
)
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