diff --git a/tests/test_superresolution.py b/tests/test_superresolution.py index 4229c449f842e07d42362e5bb7e7150f91d645f8..2b2bc2b8e1c3002fc1929828be1bd1b57c28e35a 100644 --- a/tests/test_superresolution.py +++ b/tests/test_superresolution.py @@ -52,6 +52,7 @@ def test_torch_simple(constant_h): x, y = pystencils.fields('x,y: float32[2d]') h = pystencils.fields('h0,h1,h2,h3,h4,h5,h6,h7: float32[2d]') + @crazy def move(x, y): A = sympy.Matrix([[h[0].center, h[1].center, h[2].center], @@ -59,7 +60,6 @@ def test_torch_simple(constant_h): [h[6].center, h[7].center, 1]]) return { y.center: x.interpolated_access(ProjectiveMatrix(A) @ pystencils.x_vector(2)) - } if constant_h: @@ -129,7 +129,7 @@ def test_warp(): for _ in range(NUM_LENNAS)) scale_transform(lr_warp_vectors[0], warp_vectors[0], 10).compile()().forward(input_field=lr_warp_vectors[0], - output_field=warp_vectors[0]) + output_field=warp_vectors[0]) # for i in range(len(warp_vectors)): # scale(lr_warp_vectors[i], warp_vectors[i], 10) @@ -157,7 +157,6 @@ def test_to_polar(): lenna_file = join(dirname(__file__), "test_data", "lenna.png") lenna = skimage.io.imread(lenna_file, as_gray=True).astype(np.float32) - hr, lr = pystencils.fields('hr, lr: float32[2d]') hr.set_coordinate_origin_to_field_center() @@ -165,6 +164,7 @@ def test_to_polar(): lr.coordinate_transform = lambda x: sympy.Matrix((x.norm(), sympy.atan2(*x) / sympy.pi * 500)) + def test_polar_transform(): x, y = pystencils.fields('x, y: float32[2d]')