Pillow/Tests/test_file_xpm.py

from PIL import Image, XpmImagePlugin

from .helper import PillowTestCase, assert_image_similar, hopper

TEST_FILE = "Tests/images/hopper.xpm"


class TestFileXpm(PillowTestCase):
    def test_sanity(self):
        with Image.open(TEST_FILE) as im:
            im.load()
            self.assertEqual(im.mode, "P")
            self.assertEqual(im.size, (128, 128))
            self.assertEqual(im.format, "XPM")

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

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

        self.assertRaises(SyntaxError, XpmImagePlugin.XpmImageFile, invalid_file)

    def test_load_read(self):
        # Arrange
        with Image.open(TEST_FILE) as im:
            dummy_bytes = 1

            # Act
            data = im.load_read(dummy_bytes)

        # Assert
        self.assertEqual(len(data), 16384)