diff options
| -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() |