From 50f410c177488c7d803e52331a53853a6a6c4cd3 Mon Sep 17 00:00:00 2001 From: Andrew Murray Date: Sat, 31 Dec 2022 13:06:46 +1100 Subject: [PATCH] Only compare bytes once --- Tests/helper.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/Tests/helper.py b/Tests/helper.py index 62f07b9b6..fd97a5614 100644 --- a/Tests/helper.py +++ b/Tests/helper.py @@ -87,7 +87,10 @@ def assert_image(im, mode, size, msg=None): def assert_image_equal(a, b, msg=None): assert a.mode == b.mode, msg or f"got mode {repr(a.mode)}, expected {repr(b.mode)}" assert a.size == b.size, msg or f"got size {repr(a.size)}, expected {repr(b.size)}" - if a.tobytes() != b.tobytes(): + + try: + assert a.tobytes() == b.tobytes(), msg or "got different content" + except AssertionError: if HAS_UPLOADER: try: url = test_image_results.upload(a, b) @@ -95,7 +98,7 @@ def assert_image_equal(a, b, msg=None): except Exception: pass - assert a.tobytes() == b.tobytes(), msg or "got different content" + raise def assert_image_equal_tofile(a, filename, msg=None, mode=None):