diff --git a/tests/test_superresolution.py b/tests/test_superresolution.py index b7b6248bcb9e99eca0185e613331b1463fb67e94..16c7d2f71ec6623cf29a525fe42f6dca61ebabb9 100644 --- a/tests/test_superresolution.py +++ b/tests/test_superresolution.py @@ -119,7 +119,9 @@ def test_polar_inverted_transform(): class PolarTransform(sympy.Function): def eval(args): return sympy.Matrix( - (args.norm(), sympy.atan2(args[1] - x.shape[1] / 2, args[0] - x.shape[0] / 2) / sympy.pi * x.shape[1] / 2)) + (args.norm(), + sympy.atan2(args[1] - x.shape[1] / 2, + args[0] - x.shape[0] / 2) / sympy.pi * x.shape[1] / 2)) def inv(): return lambda l: (sympy.Matrix((sympy.cos(l[1] * sympy.pi / x.shape[1] * 2) * l[0], @@ -346,8 +348,8 @@ def test_get_shift_tensors(scalar_experiment): dh.cpu_arrays.x = lenna assignments = pystencils_reco.AssignmentCollection({ - y.center: x.interpolated_access((tx.center + pystencils.x_, 2 * ty.center + - pystencils.y_)) + y.center: x.interpolated_access((tx.center + pystencils.x_, + 2 * ty.center + pystencils.y_)) }) print(pystencils.autodiff.create_backward_assignments(assignments))