Pillow/Tests/test_file_container.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

131 lines
3.0 KiB
Python
Raw Normal View History

from __future__ import annotations
2024-01-20 14:23:03 +03:00
2022-08-23 14:41:32 +03:00
import pytest
from PIL import ContainerIO, Image
2017-03-01 12:20:18 +03:00
from .helper import hopper
2017-03-01 12:20:18 +03:00
2017-03-05 01:46:30 +03:00
TEST_FILE = "Tests/images/dummy.container"
2017-03-01 12:20:18 +03:00
def test_sanity() -> None:
dir(Image)
dir(ContainerIO)
2017-03-01 12:20:18 +03:00
2017-03-05 01:46:30 +03:00
def test_isatty() -> None:
with hopper() as im:
container = ContainerIO.ContainerIO(im, 0, 0)
assert container.isatty() is False
2024-03-01 17:47:50 +03:00
@pytest.mark.parametrize(
"mode, expected_value",
(
(0, 33),
(1, 66),
(2, 100),
),
)
def test_seek_mode(mode: int, expected_value: int) -> None:
# Arrange
with open(TEST_FILE, "rb") as fh:
container = ContainerIO.ContainerIO(fh, 22, 100)
# Act
container.seek(33, mode)
container.seek(33, mode)
# Assert
2024-03-01 17:47:50 +03:00
assert container.tell() == expected_value
2022-08-23 14:41:32 +03:00
@pytest.mark.parametrize("bytesmode", (True, False))
2024-01-31 13:55:32 +03:00
def test_read_n0(bytesmode: bool) -> None:
# Arrange
2022-08-23 14:41:32 +03:00
with open(TEST_FILE, "rb" if bytesmode else "r") as fh:
container = ContainerIO.ContainerIO(fh, 22, 100)
2022-08-23 14:41:32 +03:00
# Act
container.seek(81)
data = container.read()
2022-08-23 14:41:32 +03:00
# Assert
if bytesmode:
data = data.decode()
assert data == "7\nThis is line 8\n"
2022-08-23 14:41:32 +03:00
@pytest.mark.parametrize("bytesmode", (True, False))
2024-01-31 13:55:32 +03:00
def test_read_n(bytesmode: bool) -> None:
# Arrange
2022-08-23 14:41:32 +03:00
with open(TEST_FILE, "rb" if bytesmode else "r") as fh:
container = ContainerIO.ContainerIO(fh, 22, 100)
2022-08-23 14:41:32 +03:00
# Act
container.seek(81)
data = container.read(3)
2022-08-23 14:41:32 +03:00
# Assert
if bytesmode:
data = data.decode()
assert data == "7\nT"
2022-08-23 14:41:32 +03:00
@pytest.mark.parametrize("bytesmode", (True, False))
2024-01-31 13:55:32 +03:00
def test_read_eof(bytesmode: bool) -> None:
# Arrange
2022-08-23 14:41:32 +03:00
with open(TEST_FILE, "rb" if bytesmode else "r") as fh:
container = ContainerIO.ContainerIO(fh, 22, 100)
2022-08-23 14:41:32 +03:00
# Act
container.seek(100)
data = container.read()
2022-08-23 14:41:32 +03:00
# Assert
if bytesmode:
data = data.decode()
assert data == ""
2022-08-23 14:41:32 +03:00
@pytest.mark.parametrize("bytesmode", (True, False))
2024-01-31 13:55:32 +03:00
def test_readline(bytesmode: bool) -> None:
# Arrange
2022-08-23 14:41:32 +03:00
with open(TEST_FILE, "rb" if bytesmode else "r") as fh:
container = ContainerIO.ContainerIO(fh, 0, 120)
2022-08-23 14:41:32 +03:00
# Act
data = container.readline()
2022-08-23 14:41:32 +03:00
# Assert
if bytesmode:
data = data.decode()
assert data == "This is line 1\n"
2022-08-23 14:41:32 +03:00
@pytest.mark.parametrize("bytesmode", (True, False))
2024-01-31 13:55:32 +03:00
def test_readlines(bytesmode: bool) -> None:
# Arrange
2022-08-23 14:41:32 +03:00
expected = [
"This is line 1\n",
"This is line 2\n",
"This is line 3\n",
"This is line 4\n",
"This is line 5\n",
"This is line 6\n",
"This is line 7\n",
"This is line 8\n",
]
with open(TEST_FILE, "rb" if bytesmode else "r") as fh:
container = ContainerIO.ContainerIO(fh, 0, 120)
# Act
data = container.readlines()
# Assert
if bytesmode:
data = [line.decode() for line in data]
assert data == expected