|
1 | 1 | """Test the pypdf._page module.""" |
2 | 2 | import json |
3 | 3 | import math |
4 | | -from copy import deepcopy |
5 | 4 | from io import BytesIO |
6 | 5 | from pathlib import Path |
7 | 6 | from random import shuffle |
@@ -154,25 +153,25 @@ def test_mediabox_expansion_after_rotation( |
154 | 153 |
|
155 | 154 |
|
156 | 155 | def test_transformation_equivalence(): |
157 | | - pdf_path = RESOURCE_ROOT / "labeled-edges-center-image.pdf" |
158 | | - reader_base = PdfReader(pdf_path) |
159 | | - page_base = reader_base.pages[0] |
| 156 | + def get_pages() -> tuple[PageObject, PageObject]: |
| 157 | + pdf_path = RESOURCE_ROOT / "labeled-edges-center-image.pdf" |
| 158 | + reader_base = PdfReader(pdf_path) |
| 159 | + page_base = reader_base.pages[0] |
160 | 160 |
|
161 | | - pdf_path = RESOURCE_ROOT / "box.pdf" |
162 | | - reader_add = PdfReader(pdf_path) |
163 | | - page_box = reader_add.pages[0] |
| 161 | + pdf_path = RESOURCE_ROOT / "box.pdf" |
| 162 | + reader_add = PdfReader(pdf_path) |
| 163 | + page_box = reader_add.pages[0] |
| 164 | + return page_box, page_base |
164 | 165 |
|
165 | 166 | op = Transformation().scale(2).rotate(45) |
166 | 167 |
|
167 | 168 | # Option 1: The new way |
168 | | - page_box1 = deepcopy(page_box) |
169 | | - page_base1 = deepcopy(page_base) |
| 169 | + page_box1, page_base1 = get_pages() |
170 | 170 | page_box1.add_transformation(op, expand=True) |
171 | 171 | page_base1.merge_page(page_box1, expand=False) |
172 | 172 |
|
173 | 173 | # Option 2: The old way |
174 | | - page_box2 = deepcopy(page_box) |
175 | | - page_base2 = deepcopy(page_base) |
| 174 | + page_box2, page_base2 = get_pages() |
176 | 175 | page_base2.merge_transformed_page(page_box2, op, expand=False) |
177 | 176 | page_box2.add_transformation(op) |
178 | 177 | page_base2.merge_page(page_box2) |
|
0 commit comments