set verbosity to 0 before comparing image bytes

This commit is contained in:
Yay295 2023-01-03 00:23:45 -06:00
parent 6a3b9f540f
commit 2d1782a349
2 changed files with 19 additions and 0 deletions

View File

@ -15,6 +15,9 @@ from packaging.version import parse as parse_version
from PIL import Image, ImageMath, features
# overridden in conftest.py
pytestconfig = {"option": {"verbose": 0}}
logger = logging.getLogger(__name__)
@ -88,7 +91,10 @@ 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)}"
original_verbosity = pytestconfig.option.verbose
try:
# set pytest's verbosity to 0 so that it doesn't show the full bytes diff
pytestconfig.option.verbose = 0
assert a.tobytes() == b.tobytes(), msg or "got different content"
except AssertionError:
if HAS_UPLOADER:
@ -99,6 +105,8 @@ def assert_image_equal(a, b, msg=None):
pass
raise
finally:
pytestconfig.option.verbose = original_verbosity
def assert_image_equal_tofile(a, filename, msg=None, mode=None):

View File

@ -1 +1,12 @@
pytest_plugins = ["Tests.helper"]
def pytest_configure(config):
"""
Keep a reference to pytest's configuration in our helper class.
This function is called by pytest each time the configuration is updated.
https://docs.pytest.org/en/latest/reference/reference.html#pytest.hookspec.pytest_configure
"""
import Tests.helper
Tests.helper.pytestconfig = config