Expanded .ani and .cur functionality

Tests/test_file_ani.py

@@ -0,0 +1,171 @@
+from __future__ import annotations
+
+from io import BytesIO
+
+import pytest
+
+from PIL import Image
+
+
+def test_aero_busy():
+    with Image.open("Tests/images/ani/aero_busy.ani") as im:
+        assert im.size == (64, 64)
+        assert im.info["frames"] == 18
+
+        with Image.open("Tests/images/ani/aero_busy_0.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        im.seek(8)
+        with Image.open("Tests/images/ani/aero_busy_8.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        with pytest.raises(EOFError):
+            im.seek(-1)
+
+        with pytest.raises(EOFError):
+            im.seek(18)
+
+
+def test_posy_busy():
+    with Image.open("Tests/images/ani/posy_busy.ani") as im:
+        assert im.size == (96, 96)
+        assert im.info["frames"] == 77
+
+        with Image.open("Tests/images/ani/posy_busy_0.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        im.seek(24)
+        with Image.open("Tests/images/ani/posy_busy_24.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        with pytest.raises(EOFError):
+            im.seek(77)
+
+
+def test_stopwtch():
+    with Image.open("Tests/images/ani/stopwtch.ani") as im:
+        assert im.size == (32, 32)
+        assert im.info["frames"] == 8
+
+        assert im.info["seq"][0] == 0
+        assert im.info["seq"][2] == 0
+
+        for i, r in enumerate(im.info["rate"]):
+            if i == 1 or i == 2:
+                assert r == 16
+            else:
+                assert r == 8
+
+        with Image.open("Tests/images/ani/stopwtch_0.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        im.seek(5)
+        with Image.open("Tests/images/ani/stopwtch_5.png") as png:
+            assert png.tobytes() == im.tobytes()
+
+        with pytest.raises(EOFError):
+            im.seek(8)
+
+
+def test_save():
+    directory_path = "Tests/images/ani/"
+    filenames = [
+        "aero_busy_0.png",
+        "aero_busy_8.png",
+        "posy_busy_0.png",
+        "posy_busy_24.png",
+        "stopwtch_0.png",
+        "stopwtch_5.png",
+    ]
+
+    images = [Image.open(directory_path + filename) for filename in filenames]
+
+    with BytesIO() as output:
+        images[0].save(
+            output, append_images=[images[1]], seq=[0, 1], rate=[5, 10], format="ANI"
+        )
+
+        with Image.open(output, formats=["ANI"]) as im:
+            assert im.tobytes() == images[0].tobytes()
+            im.seek(1)
+            assert im.tobytes() == images[1].tobytes()
+            assert im.info["seq"] == [0, 1]
+            assert im.info["rate"] == [5, 10]
+
+    with BytesIO() as output:
+        images[2].save(
+            output,
+            append_images=[images[3]],
+            seq=[1, 0],
+            rate=[2, 2],
+            format="ANI",
+            sizes=[(96, 96)],
+        )
+
+        with Image.open(output, formats=["ANI"]) as im:
+            assert im.tobytes() == images[2].tobytes()
+            im.seek(1)
+            assert im.tobytes() == images[3].tobytes()
+            assert im.info["seq"] == [1, 0]
+            assert im.info["rate"] == [2, 2]
+
+    with BytesIO() as output:
+        images[4].save(
+            output, append_images=[images[5]], seq=[0, 1], rate=[3, 4], format="ANI"
+        )
+
+        with Image.open(output, formats=["ANI"]) as im:
+            assert im.tobytes() == images[4].tobytes()
+            im.seek(1)
+            assert im.tobytes() == images[5].tobytes()
+            assert im.info["seq"] == [0, 1]
+            assert im.info["rate"] == [3, 4]
+
+    with BytesIO() as output:
+        images[0].save(
+            output,
+            append_images=images[1:],
+            seq=[0, 2, 4, 1, 3, 5, 0, 1, 0, 1],
+            rate=[1, 2, 3, 1, 2, 3, 1, 2, 3, 4],
+            format="ANI",
+            sizes=[(32, 32)],
+        )
+
+        with Image.open(output, formats=["ANI"]) as im:
+            assert im.info["frames"] == 6
+            assert im.info["seq"] == [0, 2, 4, 1, 3, 5, 0, 1, 0, 1]
+            assert im.info["rate"] == [1, 2, 3, 1, 2, 3, 1, 2, 3, 4]
+            assert im.size == (32, 32)
+
+            im.seek(4)
+            assert im.tobytes() == images[4].tobytes()
+
+    with BytesIO() as output:
+        with pytest.raises(ValueError):
+            images[0].save(
+                output,
+                append_images=images[1:],
+                seq=[0, 1, 8, 1, 2],
+                rate=[1, 1, 1, 1, 1],
+                format="ANI",
+                sizes=[(32, 32)],
+            )
+
+        with pytest.raises(ValueError):
+            images[0].save(
+                output,
+                append_images=images[1:],
+                seq=[0, 1, 1, 1, 2],
+                rate=[1, 1, 1, 1],
+                format="ANI",
+                sizes=[(32, 32)],
+            )
+
+        with pytest.raises(ValueError):
+            images[0].save(
+                output,
+                append_images=images[1:],
+                rate=[1, 1, 1, 1],
+                format="ANI",
+                sizes=[(32, 32)],
+            )

Tests/test_file_cur.py

@@ -1,32 +1,135 @@
 from __future__ import annotations
 
+from io import BytesIO
+
 import pytest
 
 from PIL import CurImagePlugin, Image
 
-TEST_FILE = "Tests/images/deerstalker.cur"
-
 
-def test_sanity() -> None:
-    with Image.open(TEST_FILE) as im:
+def test_deerstalker() -> None:
+    with Image.open("Tests/images/cur/deerstalker.cur") as im:
         assert im.size == (32, 32)
+        assert im.info["hotspots"] == [(0, 0)]
         assert isinstance(im, CurImagePlugin.CurImageFile)
         # Check some pixel colors to ensure image is loaded properly
         assert im.getpixel((10, 1)) == (0, 0, 0, 0)
         assert im.getpixel((11, 1)) == (253, 254, 254, 1)
         assert im.getpixel((16, 16)) == (84, 87, 86, 255)
 
 
+def test_posy_link():
+    with Image.open("Tests/images/cur/posy_link.cur") as im:
+        assert im.size == (128, 128)
+        assert im.info["sizes"] == [(128, 128), (96, 96), (64, 64), (48, 48), (32, 32)]
+        assert im.info["hotspots"] == [(25, 7), (18, 5), (12, 3), (9, 2), (5, 1)]
+        # check some pixel colors
+        assert im.getpixel((0, 0)) == (0, 0, 0, 0)
+        assert im.getpixel((20, 20)) == (0, 0, 0, 255)
+        assert im.getpixel((40, 40)) == (255, 255, 255, 255)
+
+        im.size = (32, 32)
+        im.load()
+        assert im.getpixel((0, 0)) == (0, 0, 0, 0)
+        assert im.getpixel((10, 10)) == (191, 191, 191, 255)
+
+
+def test_stopwtch():
+    with Image.open("Tests/images/cur/stopwtch.cur") as im:
+        assert im.size == (32, 32)
+        assert im.info["hotspots"] == [(16, 19)]
+
+        assert im.getpixel((16, 16)) == (0, 0, 255, 255)
+        assert im.getpixel((8, 16)) == (255, 0, 0, 255)
+
+
+def test_win98_arrow():
+    with Image.open("Tests/images/cur/win98_arrow.cur") as im:
+        assert im.size == (32, 32)
+        assert im.info["hotspots"] == [(10, 10)]
+
+        assert im.getpixel((0, 0)) == (0, 0, 0, 0)
+        assert im.getpixel((16, 16)) == (0, 0, 0, 255)
+        assert im.getpixel((14, 19)) == (255, 255, 255, 255)
+
+
 def test_invalid_file() -> None:
-    invalid_file = "Tests/images/flower.jpg"
+    invalid_file = "Tests/images/cur/posy_link.png"
 
     with pytest.raises(SyntaxError):
         CurImagePlugin.CurImageFile(invalid_file)
 
-    no_cursors_file = "Tests/images/no_cursors.cur"
+    no_cursors_file = "Tests/images/cur/no_cursors.cur"
 
-    cur = CurImagePlugin.CurImageFile(TEST_FILE)
+    cur = CurImagePlugin.CurImageFile("Tests/images/cur/deerstalker.cur")
     cur.fp.close()
     with open(no_cursors_file, "rb") as cur.fp:
         with pytest.raises(TypeError):
             cur._open()
+
+
+def test_save_win98_arrow():
+    with Image.open("Tests/images/cur/win98_arrow.png") as im:
+        # save the data
+        with BytesIO() as output:
+            im.save(
+                output,
+                format="CUR",
+                sizes=[(32, 32)],
+                hotspots=[(10, 10)],
+                bitmap_format="bmp",
+            )
+
+            with Image.open(output) as im2:
+                assert im.tobytes() == im2.tobytes()
+
+        with BytesIO() as output:
+            im.save(output, format="CUR")
+
+            # check default save params
+            with Image.open(output) as im2:
+                assert im2.size == (32, 32)
+                assert im2.info["sizes"] == [(32, 32), (24, 24), (16, 16)]
+                assert im2.info["hotspots"] == [(0, 0), (0, 0), (0, 0)]
+
+
+def test_save_posy_link():
+    sizes = [(128, 128), (96, 96), (64, 64), (48, 48), (32, 32)]
+    hotspots = [(25, 7), (18, 5), (12, 3), (9, 2), (5, 1)]
+
+    with Image.open("Tests/images/cur/posy_link.png") as im:
+        # save the data
+        with BytesIO() as output:
+            im.save(
+                output,
+                sizes=sizes,
+                hotspots=hotspots,
+                format="CUR",
+                bitmap_format="bmp",
+            )
+
+            # make sure saved output is readable
+            # and sizes/hotspots are correct
+            with Image.open(output, formats=["CUR"]) as im2:
+                assert (128, 128) == im2.size
+                assert sizes == im2.info["sizes"]
+
+        with BytesIO() as output:
+            im.save(output, sizes=sizes[3:], hotspots=hotspots[3:], format="CUR")
+
+            # make sure saved output is readable
+            # and sizes/hotspots are correct
+            with Image.open(output, formats=["CUR"]) as im2:
+                assert (48, 48) == im2.size
+                assert sizes[3:] == im2.info["sizes"]
+
+            # make sure error is thrown when size and hotspot len's
+            # don't match
+            with pytest.raises(ValueError):
+                im.save(
+                    output,
+                    sizes=sizes[2:],
+                    hotspots=hotspots[3:],
+                    format="CUR",
+                    bitmap_format="bmp",
+                )