import pytest

from PIL import Image, QoiImagePlugin

from .helper import assert_image_equal_tofile, assert_image_similar_tofile


class TestFileQOI:
    def test_sanity(self):
        with Image.open("Tests/images/hopper.qoi") as im:
            assert im.mode == "RGB"
            assert im.size == (128, 128)
            assert im.format == "QOI"

            assert_image_equal_tofile(im, "Tests/images/hopper.png")

        with Image.open("Tests/images/pil123rgba.qoi") as im:
            assert im.mode == "RGBA"
            assert im.size == (162, 150)
            assert im.format == "QOI"

            assert_image_similar_tofile(im, "Tests/images/pil123rgba.png", 0.03)

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

        with pytest.raises(SyntaxError):
            QoiImagePlugin.QoiImageFile(invalid_file)