mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-11-04 09:57:43 +03:00
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import warnings
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, TarIO, features
|
|
|
|
from .helper import is_pypy
|
|
|
|
# Sample tar archive
|
|
TEST_TAR_FILE = "Tests/images/hopper.tar"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"codec, test_path, format",
|
|
(
|
|
("zlib", "hopper.png", "PNG"),
|
|
("jpg", "hopper.jpg", "JPEG"),
|
|
),
|
|
)
|
|
def test_sanity(codec: str, test_path: str, format: str) -> None:
|
|
if features.check(codec):
|
|
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
|
|
|
|
|
|
def test_unexpected_end(tmp_path: Path) -> None:
|
|
tmpfile = str(tmp_path / "temp.tar")
|
|
with open(tmpfile, "w"):
|
|
pass
|
|
|
|
with pytest.raises(OSError, match="unexpected end of tar file"):
|
|
with TarIO.TarIO(tmpfile, "test"):
|
|
pass
|
|
|
|
|
|
def test_cannot_find_subfile() -> None:
|
|
with pytest.raises(OSError, match="cannot find subfile"):
|
|
with TarIO.TarIO(TEST_TAR_FILE, "test"):
|
|
pass
|
|
|
|
|
|
@pytest.mark.skipif(is_pypy(), reason="Requires CPython")
|
|
def test_unclosed_file() -> None:
|
|
with pytest.warns(ResourceWarning):
|
|
TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg")
|
|
|
|
|
|
def test_close() -> None:
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("error")
|
|
|
|
tar = TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg")
|
|
tar.close()
|
|
|
|
|
|
def test_contextmanager() -> None:
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("error")
|
|
|
|
with TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg"):
|
|
pass
|