17 lines
576 B
Python
17 lines
576 B
Python
|
from helpers import absolute_sample_path
|
||
|
from pdfminer.pdfdocument import PDFDocument
|
||
|
from pdfminer.pdfpage import PDFPage
|
||
|
from pdfminer.pdfparser import PDFParser
|
||
|
|
||
|
|
||
|
class TestPdfPage(object):
|
||
|
def test_page_labels(self):
|
||
|
path = absolute_sample_path("contrib/pagelabels.pdf")
|
||
|
expected_labels = ["iii", "iv", "1", "2", "1"]
|
||
|
|
||
|
with open(path, "rb") as fp:
|
||
|
parser = PDFParser(fp)
|
||
|
doc = PDFDocument(parser)
|
||
|
for (i, page) in enumerate(PDFPage.create_pages(doc)):
|
||
|
assert page.label == expected_labels[i]
|