Switching from PyPDF2 to pypdf

chazeon · Sep 23, 2023 · 910f48e · 910f48e
1 parent 9d42a41
commit 910f48e
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 2 deletions.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -0,0 +1 @@
+pytest
diff --git a/remove.py b/remove.py
@@ -1,4 +1,4 @@
-from PyPDF2 import PdfReader, PdfWriter
+from pypdf import PdfReader, PdfWriter
 
 def removeWatermark(input_fname: str, output_fname: str):
 

diff --git a/requirements.txt b/requirements.txt
@@ -1 +1 @@
-PyPDF2
+pypdf
diff --git a/tests/lorem.pdf b/tests/lorem.pdf
diff --git a/tests/test_remove.py b/tests/test_remove.py
@@ -0,0 +1,15 @@
+import pytest
+import importlib  
+import sys
+from pathlib import Path
+
+TESTS_DIR = Path(__file__).parent
+
+from remove import removeWatermark
+
+sys.path.insert(0, str(TESTS_DIR.parent))
+
+@pytest.mark.parametrize("ifname", [TESTS_DIR / "lorem.pdf"])
+def test_removal(ifname):
+    ofname = TESTS_DIR / ("%s.out.pdf" % (Path(ifname).stem,))
+    removeWatermark(str(ifname), str(ofname))