Skip to content

Commit

Permalink
ADD test methods and outputs for pypdf powered watermarks
Browse files Browse the repository at this point in the history
  • Loading branch information
sfneal committed Jul 9, 2024
1 parent aa4b49e commit a319a98
Show file tree
Hide file tree
Showing 6 changed files with 103 additions and 0 deletions.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added tests/output/test_conduit_watermark_pypdf.pdf
Binary file not shown.
Binary file not shown.
103 changes: 103 additions & 0 deletions tests/test_conduit_watermark.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,6 +233,109 @@ def test_conduit_watermark_layered_pypdf3(self):

expected_equals_output(function_name_to_file_name(), added)

@Timer.decorator
def test_conduit_watermark_pypdf(self):
"""Apply a watermark to all pages of PDF using the `pypdf` method."""
w = Watermark(
self.pdf_path, use_receipt=False, open_file=False, tempdir=self.temp.name
)
wtrmrk = w.draw(
self.address,
str(self.town + ", " + self.state),
opacity=0.08,
rotate=self.rotate,
flatten=False,
)
added = w.add(self.pdf_path, wtrmrk, method="pypdf", suffix=None)

self.assertPdfExists(wtrmrk)
self.assertPdfExists(added)
self.assertPdfHasResources(added)

copy_pdf_to_output_directory(added, function_name_to_file_name())

@Timer.decorator
def test_conduit_watermark_underneath_pypdf(self):
"""Apply a watermark underneath original content of PDF using the `pypdf` method."""
w = Watermark(
self.pdf_path, use_receipt=False, open_file=False, tempdir=self.temp.name
)
wtrmrk = w.draw(
self.address,
str(self.town + ", " + self.state),
opacity=0.08,
rotate=self.rotate,
)
added = w.add(
self.pdf_path, wtrmrk, underneath=True, suffix=None, method="pypdf"
)

self.assertPdfExists(wtrmrk)
self.assertPdfExists(added)
self.assertPdfHasResources(added)

copy_pdf_to_output_directory(added, function_name_to_file_name())

@Timer.decorator
def test_conduit_watermark_overlay_pypdf(self):
"""Apply a watermark overlaid over original content of PDF using the `pypdf` method."""
w = Watermark(
self.pdf_path, use_receipt=False, open_file=False, tempdir=self.temp.name
)
wtrmrk = w.draw(
self.address,
str(self.town + ", " + self.state),
opacity=0.08,
rotate=self.rotate,
)
added = w.add(
self.pdf_path, wtrmrk, underneath=False, suffix=None, method="pypdf"
)

self.assertPdfExists(wtrmrk)
self.assertPdfExists(added)
self.assertPdfHasResources(added)

copy_pdf_to_output_directory(added, function_name_to_file_name())

@Timer.decorator
def test_conduit_watermark_flat_pypdf(self):
"""Apply a flattened watermark to a PDF using the `pypdf` method."""
w = Watermark(
self.pdf_path, use_receipt=False, open_file=False, tempdir=self.temp.name
)
flat = w.draw(
self.address, str(self.town + ", " + self.state), opacity=0.08, flatten=True
)
added = w.add(self.pdf_path, flat, suffix=None, method="pypdf")

self.assertPdfExists(flat)
self.assertPdfExists(added)
self.assertPdfHasResources(added)

copy_pdf_to_output_directory(added, function_name_to_file_name())

@Timer.decorator
def test_conduit_watermark_layered_pypdf(self):
"""Apply a flattened watermark to a PDF using the `pypdf` method."""
w = Watermark(
self.pdf_path, use_receipt=False, open_file=False, tempdir=self.temp.name
)
layered = w.draw(
self.address,
str(self.town + ", " + self.state),
opacity=0.08,
flatten=False,
)
added = w.add(self.pdf_path, layered, suffix=None, method="pypdf")

# Assert the watermark file exists
self.assertPdfExists(layered)
self.assertPdfExists(added)
self.assertPdfHasResources(added)

copy_pdf_to_output_directory(added, function_name_to_file_name())

@Timer.decorator
def test_conduit_watermark_label(self):
"""Apply a watermark label to a PDF file."""
Expand Down

0 comments on commit a319a98

Please sign in to comment.