fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

my.pdfs: reorganize tests a bit, fix mypy

Browse source
This commit is contained in:
Dima Gerasimov 2021-03-30 20:44:49 +01:00
parent 1342c75f50
commit 62f956fb34
6 changed files with 23 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

42
tests/pdfs/test_pdfs.py
View file

@ -1,42 +0,0 @@
#!/usr/bin/env python3
import inspect
from pathlib import Path
import tempfile
from my.pdfs import get_annots, annotated_pdfs
ROOT = Path(__file__).parent.absolute()
EXPECTED_HIGHLIGHTS = set(['Since 1994, when we first began organizing web sites, we have enjoyed a rare opportunity to participate in the birth of a new discipline. ',
'And yet, unlearn we must, ',
'',
])
def test_get_annots():
"""
Test get_annots, with a real PDF file
get_annots should return a list of three Annotation objects
"""
annotations = get_annots(Path(ROOT / 'Information Architecture for the World Wide Web.pdf'))
assert len(annotations) == 3
assert set([a.highlight for a in annotations]) == EXPECTED_HIGHLIGHTS
def test_annotated_pdfs_with_filelist():
"""
Test annotated_pdfs, with a real PDF file
annotated_pdfs should return a list of one Pdf object, with three Annotations
"""
filelist = [Path(ROOT / 'Information Architecture for the World Wide Web.pdf')]
annotations_generator = annotated_pdfs(filelist=filelist, roots=None)
assert inspect.isgeneratorfunction(annotated_pdfs)
highlights_from_pdfs = []
for pdf_object in list(annotations_generator):
highlights_from_pdfs.extend([a.highlight for a in pdf_object.annotations])
assert len(highlights_from_pdfs) == 3
assert set(highlights_from_pdfs) == EXPECTED_HIGHLIGHTS