Pillow/Tests/test_image_load.py
2024-09-09 22:34:46 +10:00

55 lines
1.1 KiB
Python

from __future__ import annotations
import logging
import os
import pytest
from PIL import Image
from .helper import hopper
def test_sanity() -> None:
im = hopper()
px = im.load()
assert px is not None
assert px[0, 0] == (20, 20, 70)
def test_close() -> None:
im = Image.open("Tests/images/hopper.gif")
im.close()
with pytest.raises(ValueError):
im.load()
with pytest.raises(ValueError):
im.getpixel((0, 0))
def test_close_after_load(caplog: pytest.LogCaptureFixture) -> None:
im = Image.open("Tests/images/hopper.gif")
im.load()
with caplog.at_level(logging.DEBUG):
im.close()
assert len(caplog.records) == 0
def test_contextmanager() -> None:
fn = None
with Image.open("Tests/images/hopper.gif") as im:
assert im.fp is not None
fn = im.fp.fileno()
os.fstat(fn)
with pytest.raises(OSError):
os.fstat(fn)
def test_contextmanager_non_exclusive_fp() -> None:
with open("Tests/images/hopper.gif", "rb") as fp:
with Image.open(fp):
pass
assert not fp.closed