Skip to content

Commit 0c839a2

Browse files
committed
improve test scripts
`show_pdf_page` and `insert_image` are now tested with rotated insertions.
1 parent 1818b02 commit 0c839a2

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

tests/test_insertimage.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"""
2-
* Insert same image in two places of a page.
2+
* Insert same image with different rotations in two places of a page.
33
* Extract bboxes and transformation matrices via different methods
44
* Assert equality of bboxes and transformations
55
* Assert image locations are inside given rectangles
@@ -19,7 +19,7 @@ def test_insert():
1919
r1 = fitz.Rect(50, 50, 100, 100)
2020
r2 = fitz.Rect(50, 150, 200, 400)
2121
page.insert_image(r1, filename=imgfile)
22-
page.insert_image(r2, filename=imgfile)
22+
page.insert_image(r2, filename=imgfile, rotate=270)
2323
imglist = page.get_images(True)
2424
assert len(imglist) == 2
2525
bbox1, transform1 = page.get_image_bbox(imglist[0], transform=True)

tests/test_showpdfpage.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
"""
22
Tests:
33
* Convert some image to a PDF
4-
* Insert it in given rectangle of a PDF page
4+
* Insert it rotated in some rectangle of a PDF page
55
* Assert PDF Form XObject has been created
6-
* Assert inserted PDF is inside given retangle
6+
* Assert that image contained in inserted PDF is inside given retangle
77
"""
88
import os
99

@@ -16,16 +16,16 @@
1616
def test_insert():
1717
doc = fitz.open()
1818
page = doc.new_page()
19-
r1 = fitz.Rect(50, 50, 100, 100) # insert in here
19+
rect = fitz.Rect(50, 50, 100, 100) # insert in here
2020
img = fitz.open(imgfile) # open image
2121
tobytes = img.convert_to_pdf() # get its PDF version (bytes object)
2222
src = fitz.open("pdf", tobytes) # open as PDF
23-
page.show_pdf_page(r1, src, 0) # insert in rectangle
24-
img = page.get_images(True)[0] # make full image list of the page
25-
assert img[-1] > 0 # xref of Form XObject!
23+
xref = page.show_pdf_page(rect, src, 0, rotate=-23) # insert in rectangle
24+
# extract just inserted image info
25+
img = page.get_images(True)[0]
26+
assert img[-1] == xref # xref of Form XObject!
2627
img = page.get_image_info()[0] # read the page's images
2728

28-
# Multiple computations may lead to rounding issues, so we need
29-
# some generosity here:
30-
bbox = list(map(round, img["bbox"]))
31-
assert bbox in r1
29+
# Multiple computations may have lead to rounding deviations, so we need
30+
# some generosity here: enlarge rect by 1 point in each direction.
31+
assert img["bbox"] in rect + (-1, -1, 1, 1)

0 commit comments

Comments
 (0)