diff --git a/tests/test_superresolution.py b/tests/test_superresolution.py index 0e6c671119705ed5db1a9a42de072cd8bf378b7f..322c2051a2567cd2c65b7a364f7655874926679c 100644 --- a/tests/test_superresolution.py +++ b/tests/test_superresolution.py @@ -11,9 +11,10 @@ from os.path import dirname, join import numpy as np import skimage.io +import sympy import pystencils -from pystencils_reco.resampling import downsample, scale_transform, translate, upsample +from pystencils_reco.resampling import downsample, resample, scale_transform, translate try: import pyconrad.autoinit @@ -62,3 +63,20 @@ def test_warp(): for i in range(len(warped)): warp_kernel(lenna[i], warped[i], warp_vectors[i]) + + +def test_polar_transform(): + x, y = pystencils.fields('x, y: float32[2d]') + + x.set_coordinate_origin_to_field_center() + y.set_coordinate_origin_to_field_center() + y.coordinate_transform = lambda x: sympy.Matrix((x.norm(), sympy.atan2(*x) / (2 * sympy.pi) * y.shape[1])) + + lenna_file = join(dirname(__file__), "test_data", "lenna.png") + lenna = skimage.io.imread(lenna_file, as_gray=True).astype(np.float32) + + transformed = np.zeros((500, 500), np.float32) + + resample(x, y).compile()(x=lenna, y=transformed) + + pyconrad.imshow(transformed)