Skip to content

Commit 14d221d

Browse files
committed
Fixed bug in affine get_params test
1 parent bdc9e7c commit 14d221d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

test/test_prototype_transforms.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -525,8 +525,10 @@ def test__get_params(self, degrees, translate, scale, shear, mocker):
525525
assert degrees[0] <= params["angle"] <= degrees[1]
526526

527527
if translate is not None:
528-
assert -translate[0] * w <= params["translations"][0] <= translate[0] * w
529-
assert -translate[1] * h <= params["translations"][1] <= translate[1] * h
528+
w_max = int(round(translate[0] * w))
529+
h_max = int(round(translate[1] * h))
530+
assert -w_max <= params["translations"][0] <= w_max
531+
assert -h_max <= params["translations"][1] <= h_max
530532
else:
531533
assert params["translations"] == (0, 0)
532534

0 commit comments

Comments
 (0)