Skip to content

Commit

Permalink
TST: Simplify test_get_images (#2668)
Browse files Browse the repository at this point in the history
* Remove writing to file
* Add separate test function for negative index
  • Loading branch information
tibor-reiss committed May 24, 2024
1 parent 91f20b2 commit 12e4047
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 19 deletions.
13 changes: 13 additions & 0 deletions tests/test_page.py
Original file line number Diff line number Diff line change
Expand Up @@ -1425,3 +1425,16 @@ def test_missing_basefont_in_type3():
name = "missing-base-font.pdf"
reader = PdfReader(BytesIO(get_data_from_url(url, name=name)))
reader.pages[0]._get_fonts()


def test_invalid_index():
src_abs = RESOURCE_ROOT / "git.pdf"
reader = PdfReader(src_abs)
with pytest.raises(TypeError):
_ = reader.pages["0"]


def test_negative_index():
src_abs = RESOURCE_ROOT / "git.pdf"
reader = PdfReader(src_abs)
assert reader.pages[0] == reader.pages[-1]
24 changes: 5 additions & 19 deletions tests/test_reader.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,30 +222,16 @@ def test_get_images(src, expected_images):

src_abs = RESOURCE_ROOT / src
reader = PdfReader(src_abs)

with pytest.raises(TypeError):
page = reader.pages["0"]

page = reader.pages[-1]
page = reader.pages[0]

images_extracted = page.images

assert len(images_extracted) == len(expected_images)
for image, expected_image in zip(images_extracted, expected_images):
assert image.name == expected_image
try:
fn = f"{src}-test-out-{image.name}"
with open(fn, "wb") as fp:
fp.write(image.data)
assert (
image.name.split(".")[-1].upper()
== Image.open(io.BytesIO(image.data)).format
)
finally:
try:
Path(fn).unlink()
except Exception:
pass
assert (
image.name.split(".")[-1].upper()
== Image.open(io.BytesIO(image.data)).format
)


@pytest.mark.parametrize(
Expand Down

0 comments on commit 12e4047

Please sign in to comment.