Pillow/Tests/test_file_mpo.py

from io import BytesIO

import pytest
from PIL import Image

from .helper import assert_image_similar, is_pypy

test_files = ["Tests/images/sugarshack.mpo", "Tests/images/frozenpond.mpo"]


def setup_module():
    codecs = dir(Image.core)
    if "jpeg_encoder" not in codecs or "jpeg_decoder" not in codecs:
        pytest.skip("jpeg support not available")


def frame_roundtrip(im, **options):
    # Note that for now, there is no MPO saving functionality
    out = BytesIO()
    im.save(out, "MPO", **options)
    test_bytes = out.tell()
    out.seek(0)
    im = Image.open(out)
    im.bytes = test_bytes  # for testing only
    return im


def test_sanity():
    for test_file in test_files:
        with Image.open(test_file) as im:
            im.load()
            assert im.mode == "RGB"
            assert im.size == (640, 480)
            assert im.format == "MPO"


@pytest.mark.skipif(is_pypy(), reason="Requires CPython")
def test_unclosed_file():
    def open():
        im = Image.open(test_files[0])
        im.load()

    pytest.warns(ResourceWarning, open)


def test_closed_file():
    def open():
        im = Image.open(test_files[0])
        im.load()
        im.close()

    pytest.warns(None, open)


def test_context_manager():
    def open():
        with Image.open(test_files[0]) as im:
            im.load()

    pytest.warns(None, open)


def test_app():
    for test_file in test_files:
        # Test APP/COM reader (@PIL135)
        with Image.open(test_file) as im:
            assert im.applist[0][0] == "APP1"
            assert im.applist[1][0] == "APP2"
            assert (
                im.applist[1][1][:16]
                == b"MPF\x00MM\x00*\x00\x00\x00\x08\x00\x03\xb0\x00"
            )
            assert len(im.applist) == 2


def test_exif():
    for test_file in test_files:
        with Image.open(test_file) as im:
            info = im._getexif()
            assert info[272] == "Nintendo 3DS"
            assert info[296] == 2
            assert info[34665] == 188


def test_frame_size():
    # This image has been hexedited to contain a different size
    # in the EXIF data of the second frame
    with Image.open("Tests/images/sugarshack_frame_size.mpo") as im:
        assert im.size == (640, 480)

        im.seek(1)
        assert im.size == (680, 480)


def test_parallax():
    # Nintendo
    with Image.open("Tests/images/sugarshack.mpo") as im:
        exif = im.getexif()
        assert exif.get_ifd(0x927C)[0x1101]["Parallax"] == -44.798187255859375

    # Fujifilm
    with Image.open("Tests/images/fujifilm.mpo") as im:
        im.seek(1)
        exif = im.getexif()
        assert exif.get_ifd(0x927C)[0xB211] == -3.125


def test_mp():
    for test_file in test_files:
        with Image.open(test_file) as im:
            mpinfo = im._getmp()
            assert mpinfo[45056] == b"0100"
            assert mpinfo[45057] == 2


def test_mp_offset():
    # This image has been manually hexedited to have an IFD offset of 10
    # in APP2 data, in contrast to normal 8
    with Image.open("Tests/images/sugarshack_ifd_offset.mpo") as im:
        mpinfo = im._getmp()
        assert mpinfo[45056] == b"0100"
        assert mpinfo[45057] == 2


def test_mp_no_data():
    # This image has been manually hexedited to have the second frame
    # beyond the end of the file
    with Image.open("Tests/images/sugarshack_no_data.mpo") as im:
        with pytest.raises(ValueError):
            im.seek(1)


def test_mp_attribute():
    for test_file in test_files:
        with Image.open(test_file) as im:
            mpinfo = im._getmp()
        frameNumber = 0
        for mpentry in mpinfo[45058]:
            mpattr = mpentry["Attribute"]
            if frameNumber:
                assert not mpattr["RepresentativeImageFlag"]
            else:
                assert mpattr["RepresentativeImageFlag"]
            assert not mpattr["DependentParentImageFlag"]
            assert not mpattr["DependentChildImageFlag"]
            assert mpattr["ImageDataFormat"] == "JPEG"
            assert mpattr["MPType"] == "Multi-Frame Image: (Disparity)"
            assert mpattr["Reserved"] == 0
            frameNumber += 1


def test_seek():
    for test_file in test_files:
        with Image.open(test_file) as im:
            assert im.tell() == 0
            # prior to first image raises an error, both blatant and borderline
            with pytest.raises(EOFError):
                im.seek(-1)
            with pytest.raises(EOFError):
                im.seek(-523)
            # after the final image raises an error,
            # both blatant and borderline
            with pytest.raises(EOFError):
                im.seek(2)
            with pytest.raises(EOFError):
                im.seek(523)
            # bad calls shouldn't change the frame
            assert im.tell() == 0
            # this one will work
            im.seek(1)
            assert im.tell() == 1
            # and this one, too
            im.seek(0)
            assert im.tell() == 0


def test_n_frames():
    with Image.open("Tests/images/sugarshack.mpo") as im:
        assert im.n_frames == 2
        assert im.is_animated


def test_eoferror():
    with Image.open("Tests/images/sugarshack.mpo") as im:
        n_frames = im.n_frames

        # Test seeking past the last frame
        with pytest.raises(EOFError):
            im.seek(n_frames)
        assert im.tell() < n_frames

        # Test that seeking to the last frame does not raise an error
        im.seek(n_frames - 1)


def test_image_grab():
    for test_file in test_files:
        with Image.open(test_file) as im:
            assert im.tell() == 0
            im0 = im.tobytes()
            im.seek(1)
            assert im.tell() == 1
            im1 = im.tobytes()
            im.seek(0)
            assert im.tell() == 0
            im02 = im.tobytes()
            assert im0 == im02
            assert im0 != im1


def test_save():
    # Note that only individual frames can be saved at present
    for test_file in test_files:
        with Image.open(test_file) as im:
            assert im.tell() == 0
            jpg0 = frame_roundtrip(im)
            assert_image_similar(im, jpg0, 30)
            im.seek(1)
            assert im.tell() == 1
            jpg1 = frame_roundtrip(im)
            assert_image_similar(im, jpg1, 30)