diff tests/test_imagemasks.py @ 3:2c135c81b16c

MERGE: upstream PyMuPDF 1.26.4 with MuPDF 1.26.7
author Franz Glasner <fzglas.hg@dom66.de>
date Mon, 15 Sep 2025 11:44:09 +0200
parents 1d09e1dec1d9
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_imagemasks.py	Mon Sep 15 11:44:09 2025 +0200
@@ -0,0 +1,31 @@
+"""
+Confirm image mask detection in TextPage extractions. 
+"""
+
+import os
+
+import pymupdf
+
+scriptdir = os.path.abspath(os.path.dirname(__file__))
+filename1 = os.path.join(scriptdir, "resources", "img-regular.pdf")
+filename2 = os.path.join(scriptdir, "resources", "img-transparent.pdf")
+
+
+def test_imagemask1():
+    doc = pymupdf.open(filename1)
+    page = doc[0]
+    blocks = page.get_text("dict")["blocks"]
+    img = blocks[0]
+    assert img["mask"] is None
+    img = page.get_image_info()[0]
+    assert img["has-mask"] is False
+
+
+def test_imagemask2():
+    doc = pymupdf.open(filename2)
+    page = doc[0]
+    blocks = page.get_text("dict")["blocks"]
+    img = blocks[0]
+    assert type(img["mask"]) is bytes
+    img = page.get_image_info()[0]
+    assert img["has-mask"] is True