Pillow

1
from __future__ import annotations
2

3
import pytest
4

5
from PIL import Image, XpmImagePlugin
6

7
from .helper import assert_image_similar, hopper
8

9
TEST_FILE = "Tests/images/hopper.xpm"
10

11

12
def test_sanity() -> None:
13
    with Image.open(TEST_FILE) as im:
14
        im.load()
15
        assert im.mode == "P"
16
        assert im.size == (128, 128)
17
        assert im.format == "XPM"
18

19
        # large error due to quantization->44 colors.
20
        assert_image_similar(im.convert("RGB"), hopper("RGB"), 60)
21

22

23
def test_invalid_file() -> None:
24
    invalid_file = "Tests/images/flower.jpg"
25

26
    with pytest.raises(SyntaxError):
27
        XpmImagePlugin.XpmImageFile(invalid_file)
28

29

30
def test_load_read() -> None:
31
    # Arrange
32
    with Image.open(TEST_FILE) as im:
33
        dummy_bytes = 1
34

35
        # Act
36
        data = im.load_read(dummy_bytes)
37

38
    # Assert
39
    assert len(data) == 16384
40