Pillow/Tests/test_file_gd.py

import pytest

from PIL import GdImageFile, UnidentifiedImageError

TEST_GD_FILE = "Tests/images/hopper.gd"


def test_sanity():
    with GdImageFile.open(TEST_GD_FILE) as im:
        assert im.size == (128, 128)
        assert im.format == "GD"


def test_bad_mode():
    with pytest.raises(ValueError):
        GdImageFile.open(TEST_GD_FILE, "bad mode")


def test_invalid_file():
    invalid_file = "Tests/images/flower.jpg"

    with pytest.raises(UnidentifiedImageError):
        GdImageFile.open(invalid_file)