Only compare bytes once

This commit is contained in:
Andrew Murray 2022-12-31 13:06:46 +11:00 committed by Yay295
parent 3f045ea194
commit 50f410c177

View File

@ -87,7 +87,10 @@ def assert_image(im, mode, size, msg=None):
def assert_image_equal(a, b, 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.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)}" 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: if HAS_UPLOADER:
try: try:
url = test_image_results.upload(a, b) url = test_image_results.upload(a, b)
@ -95,7 +98,7 @@ def assert_image_equal(a, b, msg=None):
except Exception: except Exception:
pass pass
assert a.tobytes() == b.tobytes(), msg or "got different content" raise
def assert_image_equal_tofile(a, filename, msg=None, mode=None): def assert_image_equal_tofile(a, filename, msg=None, mode=None):