Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion tests/test_rotate.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,8 @@ def test_correct_results(self, angle, keep_size, mode, padding_mode, align_corne
)
)
expected = np.stack(expected).astype(np.float32)
np.testing.assert_allclose(expected, rotated, atol=1e-1)
n_good = np.sum(np.isclose(expected, rotated, atol=1e-3))
self.assertLessEqual(expected.size - n_good, 5, "diff at most 5 pixels")

@parameterized.expand(TEST_CASES_SHAPE_3D)
def test_correct_shape(self, angle, mode, padding_mode, align_corners):
Expand Down
3 changes: 2 additions & 1 deletion tests/test_rotated.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,8 @@ def test_correct_results(self, angle, keep_size, mode, padding_mode, align_corne
expected = scipy.ndimage.rotate(
self.imt[0, 0], -np.rad2deg(angle), (0, 1), not keep_size, order=_order, mode=_mode, prefilter=False
)
np.testing.assert_allclose(expected, rotated["img"][0], atol=1e-3)
good = np.sum(np.isclose(expected, rotated["img"][0], atol=1e-3))
self.assertLessEqual(np.abs(good - expected.size), 5, "diff at most 5 pixels")

expected = scipy.ndimage.rotate(
self.segn[0, 0], -np.rad2deg(angle), (0, 1), not keep_size, order=0, mode=_mode, prefilter=False
Expand Down
3 changes: 2 additions & 1 deletion tests/test_timedcall.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,14 @@
# limitations under the License.

import multiprocessing
import sys
import time
import unittest

from tests.utils import TimedCall


@TimedCall(seconds=10, force_quit=False)
@TimedCall(seconds=10 if sys.platform == "linux" else 60, force_quit=False)
Comment thread
wyli marked this conversation as resolved.
def case_1_seconds(arg=None):
time.sleep(1)
return "good" if not arg else arg
Expand Down