mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 03:18:57 +03:00
146 lines
4.1 KiB
Python
146 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
import io
|
|
import re
|
|
from typing import Callable
|
|
|
|
import pytest
|
|
|
|
from PIL import features
|
|
|
|
from .helper import skip_unless_feature
|
|
|
|
try:
|
|
from PIL import _webp
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def test_check() -> None:
|
|
# Check the correctness of the convenience function
|
|
for module in features.modules:
|
|
assert features.check_module(module) == features.check(module)
|
|
for codec in features.codecs:
|
|
assert features.check_codec(codec) == features.check(codec)
|
|
for feature in features.features:
|
|
assert features.check_feature(feature) == features.check(feature)
|
|
|
|
|
|
def test_version() -> None:
|
|
# Check the correctness of the convenience function
|
|
# and the format of version numbers
|
|
|
|
def test(name: str, function: Callable[[str], bool]) -> None:
|
|
version = features.version(name)
|
|
if not features.check(name):
|
|
assert version is None
|
|
else:
|
|
assert function(name) == version
|
|
if name != "PIL":
|
|
assert version is None or re.search(r"\d+(\.\d+)*$", version)
|
|
|
|
for module in features.modules:
|
|
test(module, features.version_module)
|
|
for codec in features.codecs:
|
|
test(codec, features.version_codec)
|
|
for feature in features.features:
|
|
test(feature, features.version_feature)
|
|
|
|
|
|
@skip_unless_feature("webp")
|
|
def test_webp_transparency() -> None:
|
|
assert features.check("transp_webp") != _webp.WebPDecoderBuggyAlpha()
|
|
assert features.check("transp_webp") == _webp.HAVE_TRANSPARENCY
|
|
|
|
|
|
@skip_unless_feature("webp")
|
|
def test_webp_mux() -> None:
|
|
assert features.check("webp_mux") == _webp.HAVE_WEBPMUX
|
|
|
|
|
|
@skip_unless_feature("webp")
|
|
def test_webp_anim() -> None:
|
|
assert features.check("webp_anim") == _webp.HAVE_WEBPANIM
|
|
|
|
|
|
@skip_unless_feature("libjpeg_turbo")
|
|
def test_libjpeg_turbo_version() -> None:
|
|
assert re.search(r"\d+\.\d+\.\d+$", features.version("libjpeg_turbo"))
|
|
|
|
|
|
@skip_unless_feature("libimagequant")
|
|
def test_libimagequant_version() -> None:
|
|
assert re.search(r"\d+\.\d+\.\d+$", features.version("libimagequant"))
|
|
|
|
|
|
@pytest.mark.parametrize("feature", features.modules)
|
|
def test_check_modules(feature: str) -> None:
|
|
assert features.check_module(feature) in [True, False]
|
|
|
|
|
|
@pytest.mark.parametrize("feature", features.codecs)
|
|
def test_check_codecs(feature: str) -> None:
|
|
assert features.check_codec(feature) in [True, False]
|
|
|
|
|
|
def test_check_warns_on_nonexistent() -> None:
|
|
with pytest.warns(UserWarning) as cm:
|
|
has_feature = features.check("typo")
|
|
assert has_feature is False
|
|
assert str(cm[-1].message) == "Unknown feature 'typo'."
|
|
|
|
|
|
def test_supported_modules() -> None:
|
|
assert isinstance(features.get_supported_modules(), list)
|
|
assert isinstance(features.get_supported_codecs(), list)
|
|
assert isinstance(features.get_supported_features(), list)
|
|
assert isinstance(features.get_supported(), list)
|
|
|
|
|
|
def test_unsupported_codec() -> None:
|
|
# Arrange
|
|
codec = "unsupported_codec"
|
|
# Act / Assert
|
|
with pytest.raises(ValueError):
|
|
features.check_codec(codec)
|
|
with pytest.raises(ValueError):
|
|
features.version_codec(codec)
|
|
|
|
|
|
def test_unsupported_module() -> None:
|
|
# Arrange
|
|
module = "unsupported_module"
|
|
# Act / Assert
|
|
with pytest.raises(ValueError):
|
|
features.check_module(module)
|
|
with pytest.raises(ValueError):
|
|
features.version_module(module)
|
|
|
|
|
|
def test_pilinfo() -> None:
|
|
buf = io.StringIO()
|
|
features.pilinfo(buf)
|
|
out = buf.getvalue()
|
|
lines = out.splitlines()
|
|
assert lines[0] == "-" * 68
|
|
assert lines[1].startswith("Pillow ")
|
|
assert lines[2].startswith("Python ")
|
|
lines = lines[3:]
|
|
while lines[0].startswith(" "):
|
|
lines = lines[1:]
|
|
assert lines[0] == "-" * 68
|
|
assert lines[1].startswith("Python modules loaded from ")
|
|
assert lines[2].startswith("Binary modules loaded from ")
|
|
assert lines[3] == "-" * 68
|
|
jpeg = (
|
|
"\n"
|
|
+ "-" * 68
|
|
+ "\n"
|
|
+ "JPEG image/jpeg\n"
|
|
+ "Extensions: .jfif, .jpe, .jpeg, .jpg\n"
|
|
+ "Features: open, save\n"
|
|
+ "-" * 68
|
|
+ "\n"
|
|
)
|
|
assert jpeg in out
|