Skip to content

Commit bf07521

Browse files
committed
STY: Simplify assertions in registration tests for improved readability
1 parent a4bc020 commit bf07521

File tree

1 file changed

+4
-14
lines changed

1 file changed

+4
-14
lines changed

tests/registration/test_registration.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,7 @@ def test_rigid_body_registration_run_sucess():
8181

8282
resampled_image, _ = rigid_body_registration(img_orig, img_rot)
8383

84-
assert (
85-
np.mean(np.subtract(img_orig, resampled_image))
86-
< np.mean(img_orig)
87-
)
84+
assert np.mean(np.subtract(img_orig, resampled_image)) < np.mean(img_orig)
8885

8986

9087
@pytest.mark.parametrize(
@@ -225,10 +222,7 @@ def test_affine_registration_success():
225222

226223
resampled_image, _ = affine_registration(img_orig, img_rot)
227224

228-
assert (
229-
np.mean(np.subtract(img_orig, resampled_image))
230-
< np.mean(img_orig)
231-
)
225+
assert np.mean(np.subtract(img_orig, resampled_image)) < np.mean(img_orig)
232226

233227

234228
def test_affine_registration_raise_exception_if_fixed_image_not_numpy():
@@ -283,9 +277,7 @@ def test_affine_registration_fast_method():
283277

284278
assert isinstance(resampled_image, np.ndarray)
285279
assert resampled_image.shape == img_rot.shape
286-
assert np.mean(np.abs(img_orig - resampled_image)) < np.mean(
287-
img_orig
288-
)
280+
assert np.mean(np.abs(img_orig - resampled_image)) < np.mean(img_orig)
289281

290282

291283
def test_affine_registration_slow_method():
@@ -298,9 +290,7 @@ def test_affine_registration_slow_method():
298290

299291
assert isinstance(resampled_image, np.ndarray)
300292
assert resampled_image.shape == img_rot.shape
301-
assert np.mean(np.abs(img_orig - resampled_image)) < np.mean(
302-
img_orig
303-
)
293+
assert np.mean(np.abs(img_orig - resampled_image)) < np.mean(img_orig)
304294

305295

306296
def test_apply_transformation_success():

0 commit comments

Comments
 (0)