From 2a9333cac284ef2a863a38c716a57353890ef7cd Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Thu, 12 Nov 2020 13:47:49 +0100 Subject: [PATCH] Autoformat --- tests/test_superresolution.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_superresolution.py b/tests/test_superresolution.py index 4229c44..2b2bc2b 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]') -- GitLab