Pillow/Tests/test_file_palm.py

import os.path

import pytest

from .helper import (
    PillowTestCase,
    assert_image_equal,
    hopper,
    imagemagick_available,
    skip_known_bad_test,
)


class TestFilePalm(PillowTestCase):
    _roundtrip = imagemagick_available()

    def helper_save_as_palm(self, mode):
        # Arrange
        im = hopper(mode)
        outfile = self.tempfile("temp_" + mode + ".palm")

        # Act
        im.save(outfile)

        # Assert
        assert os.path.isfile(outfile)
        assert os.path.getsize(outfile) > 0

    def roundtrip(self, mode):
        if not self._roundtrip:
            return

        im = hopper(mode)
        outfile = self.tempfile("temp.palm")

        im.save(outfile)
        converted = self.open_withImagemagick(outfile)
        assert_image_equal(converted, im)

    def test_monochrome(self):
        # Arrange
        mode = "1"

        # Act / Assert
        self.helper_save_as_palm(mode)
        self.roundtrip(mode)

    def test_p_mode(self):
        # Arrange
        mode = "P"

        # Act / Assert
        self.helper_save_as_palm(mode)
        skip_known_bad_test("Palm P image is wrong")
        self.roundtrip(mode)

    def test_l_ioerror(self):
        # Arrange
        mode = "L"

        # Act / Assert
        with pytest.raises(IOError):
            self.helper_save_as_palm(mode)

    def test_rgb_ioerror(self):
        # Arrange
        mode = "RGB"

        # Act / Assert
        with pytest.raises(IOError):
            self.helper_save_as_palm(mode)