Python2/PyMuPDF: tests/test

annotate tests/test_toc.py @ 43:202a1f38a622

FIX: New added code snippets did not yet account for the changed return values from "_fromto()"

author	Franz Glasner <fzglas.hg@dom66.de>
date	Sat, 11 Oct 2025 17:15:08 +0200
parents	1d09e1dec1d9
children

rev	line source
1 1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	1 """
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	2 * Verify equality of generated TOCs and expected results.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	3 * Verify TOC deletion works
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	4 * Verify manipulation of single TOC item works
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	5 * Verify stability against circular TOC items
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	6 """
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	7
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	8 import os
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	9 import sys
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	10 import pymupdf
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	11 import pathlib
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	12
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	13 scriptdir = os.path.abspath(os.path.dirname(__file__))
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	14 filename = os.path.join(scriptdir, "resources", "001003ED.pdf")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	15 filename2 = os.path.join(scriptdir, "resources", "2.pdf")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	16 circular = os.path.join(scriptdir, "resources", "circular-toc.pdf")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	17 full_toc = os.path.join(scriptdir, "resources", "full_toc.txt")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	18 simple_toc = os.path.join(scriptdir, "resources", "simple_toc.txt")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	19 file_3820 = os.path.join(scriptdir, "resources", "test-3820.pdf")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	20 doc = pymupdf.open(filename)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	21
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	22
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	23 def test_simple_toc():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	24 simple_lines = open(simple_toc, "rb").read()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	25 toc = b"".join([str(t).encode() for t in doc.get_toc(True)])
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	26 assert toc == simple_lines
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	27
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	28
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	29 def test_full_toc():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	30 if not hasattr(pymupdf, "mupdf"):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	31 # Classic implementation does not have fix for this test.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	32 print(f"Not running test_full_toc on classic implementation.")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	33 return
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	34 expected_path = f"{scriptdir}/resources/full_toc.txt"
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	35 expected = pathlib.Path(expected_path).read_bytes()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	36 # Github windows x32 seems to insert \r characters; maybe something to
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	37 # do with the Python installation's line endings settings.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	38 expected = expected.decode("utf8")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	39 expected = expected.replace('\r', '')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	40 toc = "\n".join([str(t) for t in doc.get_toc(False)])
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	41 toc += "\n"
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	42 assert toc == expected
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	43
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	44
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	45 def test_erase_toc():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	46 doc.set_toc([])
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	47 assert doc.get_toc() == []
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	48
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	49
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	50 def test_replace_toc():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	51 toc = doc.get_toc(False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	52 doc.set_toc(toc)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	53
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	54
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	55 def test_setcolors():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	56 doc = pymupdf.open(filename2)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	57 toc = doc.get_toc(False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	58 for i in range(len(toc)):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	59 d = toc[i][3]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	60 d["color"] = (1, 0, 0)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	61 d["bold"] = True
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	62 d["italic"] = True
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	63 doc.set_toc_item(i, dest_dict=d)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	64
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	65 toc2 = doc.get_toc(False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	66 assert len(toc2) == len(toc)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	67
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	68 for t in toc2:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	69 d = t[3]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	70 assert d["bold"]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	71 assert d["italic"]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	72 assert d["color"] == (1, 0, 0)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	73
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	74
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	75 def test_circular():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	76 """The test file contains circular bookmarks."""
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	77 doc = pymupdf.open(circular)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	78 toc = doc.get_toc(False) # this must not loop
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	79 rebased = hasattr(pymupdf, 'mupdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	80 if rebased:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	81 wt = pymupdf.TOOLS.mupdf_warnings()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	82 assert wt == 'Bad or missing prev pointer in outline tree, repairing', \
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	83 f'{wt=}'
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	84
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	85 def test_2355():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	86
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	87 # Create a test PDF with toc.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	88 doc = pymupdf.Document()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	89 for _ in range(10):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	90 doc.new_page(doc.page_count)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	91 doc.set_toc([[1, 'test', 1], [1, 'test2', 5]])
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	92
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	93 path = 'test_2355.pdf'
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	94 doc.save(path)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	95
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	96 # Open many times
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	97 for i in range(10):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	98 with pymupdf.open(path) as new_doc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	99 new_doc.get_toc()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	100
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	101 # Open once and read many times
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	102 with pymupdf.open(path) as new_doc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	103 for i in range(10):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	104 new_doc.get_toc()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	105
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	106 def test_2788():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	107 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	108 Check handling of Document.get_toc() when toc item has kind=4.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	109 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	110 if not hasattr(pymupdf, 'mupdf'):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	111 # Classic implementation does not have fix for this test.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	112 print(f'Not running test_2788 on classic implementation.')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	113 return
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	114 path = os.path.abspath(f'{__file__}/../../tests/resources/test_2788.pdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	115 document = pymupdf.open(path)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	116 toc0 = [[1, 'page2', 2, {'kind': 4, 'xref': 14, 'page': 1, 'to': pymupdf.Point(100.0, 760.0), 'zoom': 0.0, 'nameddest': 'page.2'}]]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	117 toc1 = document.get_toc(simple=False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	118 print(f'{toc0=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	119 print(f'{toc1=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	120 assert toc1 == toc0
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	121
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	122 doc.set_toc(toc0)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	123 toc2 = document.get_toc(simple=False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	124 print(f'{toc0=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	125 print(f'{toc2=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	126 assert toc2 == toc0
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	127
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	128 # Also test Page.get_links() bugfix from #2817.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	129 for page in document:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	130 page.get_links()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	131 rebased = hasattr(pymupdf, 'mupdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	132 if rebased:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	133 wt = pymupdf.TOOLS.mupdf_warnings()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	134 assert wt == (
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	135 "syntax error: expected 'obj' keyword (0 3 ?)\n"
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	136 "trying to repair broken xref\n"
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	137 "repairing PDF document"
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	138 ), f'{wt=}'
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	139
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	140
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	141 def test_toc_count():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	142 file_in = os.path.abspath(f'{__file__}/../../tests/resources/test_toc_count.pdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	143 file_out = os.path.abspath(f'{__file__}/../../tests/test_toc_count_out.pdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	144
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	145 def get(doc):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	146 outlines = doc.xref_get_key(doc.pdf_catalog(), "Outlines")
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	147 ret = doc.xref_object(int(outlines[1].split()[0]))
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	148 return ret
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	149 print()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	150 with pymupdf.open(file_in) as doc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	151 print(f'1: {get(doc)}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	152 toc = doc.get_toc(simple=False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	153 doc.set_toc([])
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	154 #print(f'2: {get(doc)}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	155 doc.set_toc(toc)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	156 print(f'3: {get(doc)}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	157 doc.save(file_out, garbage=4)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	158 with pymupdf.open(file_out) as doc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	159 print(f'4: {get(doc)}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	160 pymupdf._log_items_clear()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	161
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	162
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	163 def test_3347():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	164 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	165 Check fix for #3347 - link destination rectangles when source/destination
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	166 pages have different sizes.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	167 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	168 doc = pymupdf.open()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	169 doc.new_page(width=500, height=800)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	170 doc.new_page(width=800, height=500)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	171 rects = [
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	172 (0, pymupdf.Rect(10, 20, 50, 40), pymupdf.utils.getColor('red')),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	173 (0, pymupdf.Rect(300, 350, 400, 450), pymupdf.utils.getColor('green')),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	174 (1, pymupdf.Rect(20, 30, 40, 50), pymupdf.utils.getColor('blue')),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	175 (1, pymupdf.Rect(350, 300, 450, 400), pymupdf.utils.getColor('black'))
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	176 ]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	177
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	178 for page, rect, color in rects:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	179 doc[page].draw_rect(rect, color=color)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	180
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	181 for (from_page, from_rect, _), (to_page, to_rect, _) in zip(rects, rects[1:] + rects[:1]):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	182 doc[from_page].insert_link({
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	183 'kind': 1,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	184 'from': from_rect,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	185 'page': to_page,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	186 'to': to_rect.top_left,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	187 })
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	188
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	189 links_expected = [
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	190 (0, {'kind': 1, 'xref': 11, 'from': pymupdf.Rect(10.0, 20.0, 50.0, 40.0), 'page': 0, 'to': pymupdf.Point(300.0, 350.0), 'zoom': 0.0, 'id': 'fitz-L0'}),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	191 (0, {'kind': 1, 'xref': 12, 'from': pymupdf.Rect(300.0, 350.0, 400.0, 450.0), 'page': 1, 'to': pymupdf.Point(20.0, 30.0), 'zoom': 0.0, 'id': 'fitz-L1'}),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	192 (1, {'kind': 1, 'xref': 13, 'from': pymupdf.Rect(20.0, 30.0, 40.0, 50.0), 'page': 1, 'to': pymupdf.Point(350.0, 300.0), 'zoom': 0.0, 'id': 'fitz-L0'}),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	193 (1, {'kind': 1, 'xref': 14, 'from': pymupdf.Rect(350.0, 300.0, 450.0, 400.0), 'page': 0, 'to': pymupdf.Point(10.0, 20.0), 'zoom': 0.0, 'id': 'fitz-L1'}),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	194 ]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	195
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	196 path = os.path.normpath(f'{__file__}/../../tests/test_3347_out.pdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	197 doc.save(path)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	198 print(f'Have saved to {path=}.')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	199
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	200 links_actual = list()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	201 for page_i, page in enumerate(doc):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	202 links = page.get_links()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	203 for link_i, link in enumerate(links):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	204 print(f'{page_i=} {link_i=}: {link!r}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	205 links_actual.append( (page_i, link) )
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	206
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	207 assert links_actual == links_expected
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	208
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	209
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	210 def test_3400():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	211 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	212 Check fix for #3400 - link destination rectangles when source/destination
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	213 pages have different rotations.
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	214 '''
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	215 width = 750
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	216 height = 1110
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	217 circle_middle_point = pymupdf.Point(height / 4, width / 4)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	218 print(f'{circle_middle_point=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	219 with pymupdf.open() as doc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	220
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	221 page = doc.new_page(width=width, height=height)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	222 page.set_rotation(270)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	223 # draw a circle at the middle point to facilitate debugging
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	224 page.draw_circle(circle_middle_point, color=(0, 0, 1), radius=5, width=2)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	225
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	226 for i in range(10):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	227 for j in range(10):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	228 x = i/10 * width
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	229 y = j/10 * height
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	230 page.draw_circle(pymupdf.Point(x, y), color=(0,0,0), radius=0.2, width=0.1)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	231 page.insert_htmlbox(pymupdf.Rect(x, y, x+width/10, y+height/20), f'<small><small><small><small>({x=:.1f},{y=:.1f})</small></small></small></small>', )
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	232
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	233 # rotate the middle point by the page rotation for the new toc entry
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	234 toc_link_coords = circle_middle_point
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	235 print(f'{toc_link_coords=}')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	236
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	237 toc = [
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	238 (
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	239 1,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	240 "Link to circle",
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	241 1,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	242 {
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	243 "kind": pymupdf.LINK_GOTO,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	244 "page": 1,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	245 "to": toc_link_coords,
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	246 "from": pymupdf.Rect(0, 0, height / 4, width / 4),
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	247 },
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	248 )
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	249 ]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	250 doc.set_toc(toc, 0) # set the toc
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	251
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	252 page = doc.new_page(width=200, height=300)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	253 from_rect = pymupdf.Rect(10, 10, 100, 50)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	254 page.insert_htmlbox(from_rect, 'link')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	255 link = dict()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	256 link['from'] = from_rect
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	257 link['kind'] = pymupdf.LINK_GOTO
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	258 link['to'] = toc_link_coords
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	259 link['page'] = 0
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	260 page.insert_link(link)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	261
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	262 path = os.path.normpath(f'{__file__}/../../tests/test_3400.pdf')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	263 doc.save(path)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	264 print(f'Saved to {path=}.')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	265
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	266 links_expected = [
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	267 (1, {'kind': 1, 'xref': 1120, 'from': pymupdf.Rect(10.0, 10.0, 100.0, 50.0), 'page': 0, 'to': pymupdf.Point(187.5, 472.5), 'zoom': 0.0, 'id': 'fitz-L0'})
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	268 ]
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	269
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	270 links_actual = list()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	271 for page_i, page in enumerate(doc):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	272 links = page.get_links()
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	273 for link_i, link in enumerate(links):
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	274 print(f'({page_i}, {link!r})')
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	275 links_actual.append( (page_i, link) )
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	276
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	277 assert links_actual == links_expected
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	278
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	279
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	280
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	281 def test_3820():
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	282 """Ensure all extended TOC items point to pages."""
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	283 doc = pymupdf.open(file_3820)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	284 toc = doc.get_toc(simple=False)
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	285 for _, _, epage, dest in toc:
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	286 assert epage == dest["page"] + 1
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	287
1d09e1dec1d9 ADD: PyMuPDF v1.26.4: the original sdist. Franz Glasner <fzglas.hg@dom66.de> parents: diff changeset	288

Mercurial > hgrepos > Python2 > PyMuPDF

annotate tests/test_toc.py @ 43:202a1f38a622