diff options
| author | Jocelyn Boullier <jocelyn@boullier.bzh> | 2021-03-02 10:14:52 +0100 |
|---|---|---|
| committer | Jocelyn Boullier <jocelyn@boullier.bzh> | 2021-03-02 21:35:03 +0100 |
| commit | ee2d05d88d1972d56e93f8c179690fe3a6e308a3 (patch) | |
| tree | 97a32432194152f18261e27e9759e3c649d755d8 | |
| parent | feat: use pikepdf instead of PyPDF2, bumps Python min. version to 3.6 (diff) | |
| download | paper2remarkable-ee2d05d88d1972d56e93f8c179690fe3a6e308a3.tar.gz paper2remarkable-ee2d05d88d1972d56e93f8c179690fe3a6e308a3.zip | |
tests: add test for keeping the ToC after processing
| -rw-r--r-- | tests/test_providers.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/test_providers.py b/tests/test_providers.py index eaeb8aa..9eec83b 100644 --- a/tests/test_providers.py +++ b/tests/test_providers.py @@ -11,6 +11,7 @@ import pdfplumber import shutil import tempfile import unittest +from pikepdf import Pdf from paper2remarkable.providers import ( ACL, @@ -34,6 +35,7 @@ from paper2remarkable.providers import ( Springer, TandFOnline, ) +from paper2remarkable.utils import download_url VERBOSE = False @@ -438,6 +440,16 @@ class TestProviders(unittest.TestCase): filename = prov.run(url) self.assertEqual(exp, os.path.basename(filename)) + def test_local_file_copy_toc(self): + """Make sure the table of content is kept after processing.""" + local_filename = "test.pdf" + download_url("https://arxiv.org/pdf/1711.03512.pdf", local_filename) + prov = LocalFile(upload=False, verbose=VERBOSE) + filename = prov.run(local_filename) + with Pdf.open(filename) as pdf: + with pdf.open_outline() as outline: + assert len(outline.root) > 0 + if __name__ == "__main__": unittest.main() |