Pillow/Tests/test_file_eps.py

import io
import unittest

import pytest
from PIL import EpsImagePlugin, Image, features

from .helper import PillowTestCase, assert_image_similar, hopper, skip_unless_feature

HAS_GHOSTSCRIPT = EpsImagePlugin.has_ghostscript()

# Our two EPS test files (they are identical except for their bounding boxes)
file1 = "Tests/images/zero_bb.eps"
file2 = "Tests/images/non_zero_bb.eps"

# Due to palletization, we'll need to convert these to RGB after load
file1_compare = "Tests/images/zero_bb.png"
file1_compare_scale2 = "Tests/images/zero_bb_scale2.png"

file2_compare = "Tests/images/non_zero_bb.png"
file2_compare_scale2 = "Tests/images/non_zero_bb_scale2.png"

# EPS test files with binary preview
file3 = "Tests/images/binary_preview_map.eps"


class TestFileEps(PillowTestCase):
    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_sanity(self):
        # Regular scale
        with Image.open(file1) as image1:
            image1.load()
            assert image1.mode == "RGB"
            assert image1.size == (460, 352)
            assert image1.format == "EPS"

        with Image.open(file2) as image2:
            image2.load()
            assert image2.mode == "RGB"
            assert image2.size == (360, 252)
            assert image2.format == "EPS"

        # Double scale
        with Image.open(file1) as image1_scale2:
            image1_scale2.load(scale=2)
            assert image1_scale2.mode == "RGB"
            assert image1_scale2.size == (920, 704)
            assert image1_scale2.format == "EPS"

        with Image.open(file2) as image2_scale2:
            image2_scale2.load(scale=2)
            assert image2_scale2.mode == "RGB"
            assert image2_scale2.size == (720, 504)
            assert image2_scale2.format == "EPS"

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

        with pytest.raises(SyntaxError):
            EpsImagePlugin.EpsImageFile(invalid_file)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_cmyk(self):
        with Image.open("Tests/images/pil_sample_cmyk.eps") as cmyk_image:

            assert cmyk_image.mode == "CMYK"
            assert cmyk_image.size == (100, 100)
            assert cmyk_image.format == "EPS"

            cmyk_image.load()
            assert cmyk_image.mode == "RGB"

            if features.check("jpg"):
                with Image.open("Tests/images/pil_sample_rgb.jpg") as target:
                    assert_image_similar(cmyk_image, target, 10)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_showpage(self):
        # See https://github.com/python-pillow/Pillow/issues/2615
        with Image.open("Tests/images/reqd_showpage.eps") as plot_image:
            with Image.open("Tests/images/reqd_showpage.png") as target:
                # should not crash/hang
                plot_image.load()
                #  fonts could be slightly different
                assert_image_similar(plot_image, target, 6)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_file_object(self):
        # issue 479
        with Image.open(file1) as image1:
            with open(self.tempfile("temp_file.eps"), "wb") as fh:
                image1.save(fh, "EPS")

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_iobase_object(self):
        # issue 479
        with Image.open(file1) as image1:
            with open(self.tempfile("temp_iobase.eps"), "wb") as fh:
                image1.save(fh, "EPS")

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_bytesio_object(self):
        with open(file1, "rb") as f:
            img_bytes = io.BytesIO(f.read())

        with Image.open(img_bytes) as img:
            img.load()

            with Image.open(file1_compare) as image1_scale1_compare:
                image1_scale1_compare = image1_scale1_compare.convert("RGB")
            image1_scale1_compare.load()
            assert_image_similar(img, image1_scale1_compare, 5)

    def test_image_mode_not_supported(self):
        im = hopper("RGBA")
        tmpfile = self.tempfile("temp.eps")
        with pytest.raises(ValueError):
            im.save(tmpfile)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    @skip_unless_feature("zlib")
    def test_render_scale1(self):
        # We need png support for these render test

        # Zero bounding box
        with Image.open(file1) as image1_scale1:
            image1_scale1.load()
            with Image.open(file1_compare) as image1_scale1_compare:
                image1_scale1_compare = image1_scale1_compare.convert("RGB")
            image1_scale1_compare.load()
            assert_image_similar(image1_scale1, image1_scale1_compare, 5)

        # Non-Zero bounding box
        with Image.open(file2) as image2_scale1:
            image2_scale1.load()
            with Image.open(file2_compare) as image2_scale1_compare:
                image2_scale1_compare = image2_scale1_compare.convert("RGB")
            image2_scale1_compare.load()
            assert_image_similar(image2_scale1, image2_scale1_compare, 10)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    @skip_unless_feature("zlib")
    def test_render_scale2(self):
        # We need png support for these render test

        # Zero bounding box
        with Image.open(file1) as image1_scale2:
            image1_scale2.load(scale=2)
            with Image.open(file1_compare_scale2) as image1_scale2_compare:
                image1_scale2_compare = image1_scale2_compare.convert("RGB")
            image1_scale2_compare.load()
            assert_image_similar(image1_scale2, image1_scale2_compare, 5)

        # Non-Zero bounding box
        with Image.open(file2) as image2_scale2:
            image2_scale2.load(scale=2)
            with Image.open(file2_compare_scale2) as image2_scale2_compare:
                image2_scale2_compare = image2_scale2_compare.convert("RGB")
            image2_scale2_compare.load()
            assert_image_similar(image2_scale2, image2_scale2_compare, 10)

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_resize(self):
        files = [file1, file2, "Tests/images/illu10_preview.eps"]
        for fn in files:
            with Image.open(fn) as im:
                new_size = (100, 100)
                im = im.resize(new_size)
                assert im.size == new_size

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_thumbnail(self):
        # Issue #619
        # Arrange
        files = [file1, file2]
        for fn in files:
            with Image.open(file1) as im:
                new_size = (100, 100)
                im.thumbnail(new_size)
                assert max(im.size) == max(new_size)

    def test_read_binary_preview(self):
        # Issue 302
        # open image with binary preview
        with Image.open(file3):
            pass

    def _test_readline(self, t, ending):
        ending = "Failure with line ending: %s" % (
            "".join("%s" % ord(s) for s in ending)
        )
        assert t.readline().strip("\r\n") == "something", ending
        assert t.readline().strip("\r\n") == "else", ending
        assert t.readline().strip("\r\n") == "baz", ending
        assert t.readline().strip("\r\n") == "bif", ending

    def _test_readline_io_psfile(self, test_string, ending):
        f = io.BytesIO(test_string.encode("latin-1"))
        t = EpsImagePlugin.PSFile(f)
        self._test_readline(t, ending)

    def _test_readline_file_psfile(self, test_string, ending):
        f = self.tempfile("temp.txt")
        with open(f, "wb") as w:
            w.write(test_string.encode("latin-1"))

        with open(f, "rb") as r:
            t = EpsImagePlugin.PSFile(r)
            self._test_readline(t, ending)

    def test_readline(self):
        # check all the freaking line endings possible from the spec
        # test_string = u'something\r\nelse\n\rbaz\rbif\n'
        line_endings = ["\r\n", "\n", "\n\r", "\r"]
        strings = ["something", "else", "baz", "bif"]

        for ending in line_endings:
            s = ending.join(strings)
            self._test_readline_io_psfile(s, ending)
            self._test_readline_file_psfile(s, ending)

    def test_open_eps(self):
        # https://github.com/python-pillow/Pillow/issues/1104
        # Arrange
        FILES = [
            "Tests/images/illu10_no_preview.eps",
            "Tests/images/illu10_preview.eps",
            "Tests/images/illuCS6_no_preview.eps",
            "Tests/images/illuCS6_preview.eps",
        ]

        # Act / Assert
        for filename in FILES:
            with Image.open(filename) as img:
                assert img.mode == "RGB"

    @unittest.skipUnless(HAS_GHOSTSCRIPT, "Ghostscript not available")
    def test_emptyline(self):
        # Test file includes an empty line in the header data
        emptyline_file = "Tests/images/zero_bb_emptyline.eps"

        with Image.open(emptyline_file) as image:
            image.load()
        assert image.mode == "RGB"
        assert image.size == (460, 352)
        assert image.format == "EPS"