mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 19:38:57 +03:00
53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
import pytest
|
|
from PIL import Image, TarIO
|
|
|
|
from .helper import is_pypy
|
|
|
|
codecs = dir(Image.core)
|
|
|
|
# Sample tar archive
|
|
TEST_TAR_FILE = "Tests/images/hopper.tar"
|
|
|
|
|
|
def setup_module():
|
|
if "zip_decoder" not in codecs and "jpeg_decoder" not in codecs:
|
|
pytest.skip("neither jpeg nor zip support available")
|
|
|
|
|
|
def test_sanity():
|
|
for codec, test_path, format in [
|
|
["zip_decoder", "hopper.png", "PNG"],
|
|
["jpeg_decoder", "hopper.jpg", "JPEG"],
|
|
]:
|
|
if codec in codecs:
|
|
with TarIO.TarIO(TEST_TAR_FILE, test_path) as tar:
|
|
with Image.open(tar) as im:
|
|
im.load()
|
|
assert im.mode == "RGB"
|
|
assert im.size == (128, 128)
|
|
assert im.format == format
|
|
|
|
|
|
@pytest.mark.skipif(is_pypy(), reason="Requires CPython")
|
|
def test_unclosed_file():
|
|
def open():
|
|
TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg")
|
|
|
|
pytest.warns(ResourceWarning, open)
|
|
|
|
|
|
def test_close():
|
|
def open():
|
|
tar = TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg")
|
|
tar.close()
|
|
|
|
pytest.warns(None, open)
|
|
|
|
|
|
def test_contextmanager():
|
|
def open():
|
|
with TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg"):
|
|
pass
|
|
|
|
pytest.warns(None, open)
|