josch/img2pdf
3
6
Fork 10
Code Issues 22 Pull requests 5 Projects Releases 3 Wiki Activity

src/tests/__init__.py: add support for engine instead of with_pdfrw

Browse source
This commit is contained in:
Johannes 'josch' Schauer 2020-05-28 13:12:07 +02:00
parent 7b58c4e58d
commit 5c7ffb09a1
Signed by: josch
GPG key ID: F2CBA5C78FBD83E1
1 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/tests/__init__.py
View file

@ -476,23 +476,25 @@ def test_suite():
setattr(TestImg2Pdf, "test_layout_%03d_im2" % i, layout_handler_im2) setattr(TestImg2Pdf, "test_layout_%03d_im2" % i, layout_handler_im2)
files = os.listdir(os.path.join(HERE, "input")) files = os.listdir(os.path.join(HERE, "input"))
for with_pdfrw, test_name in [(a, b) for a in [True, False] for engine, test_name in [(a, b) for a in [img2pdf.Engine.internal, img2pdf.Engine.pdfrw]
for b in files]: for b in files]:
# we do not test animation.gif with pdfrw because it doesn't support
# saving hexadecimal palette data
if test_name == 'animation.gif' and with_pdfrw:
continue
inputf = os.path.join(HERE, "input", test_name) inputf = os.path.join(HERE, "input", test_name)
if not os.path.isfile(inputf): if not os.path.isfile(inputf):
continue continue
outputf = os.path.join(HERE, "output", test_name+".pdf") outputf = os.path.join(HERE, "output", test_name+".pdf")
assert os.path.isfile(outputf) assert os.path.isfile(outputf)
def handle(self, f=inputf, out=outputf, with_pdfrw=with_pdfrw): def handle(self, f=inputf, out=outputf, engine=engine):
if not img2pdf.have_pdfrw and engine == img2pdf.Engine.pdfrw:
self.skipTest("pdfrw not available")
# we do not test animation.gif with pdfrw because it doesn't support
# saving hexadecimal palette data
if f.endswith("/animation.gif") and engine == img2pdf.Engine.pdfrw:
self.skipTest("pdfrw does not support palletes")
with open(f, "rb") as inf: with open(f, "rb") as inf:
orig_imgdata = inf.read() orig_imgdata = inf.read()
output = img2pdf.convert(orig_imgdata, nodate=True, output = img2pdf.convert(orig_imgdata, nodate=True,
with_pdfrw=with_pdfrw) engine=engine)
x = pikepdf.open(BytesIO(output)) x = pikepdf.open(BytesIO(output))
self.assertIn(x.Root.Pages.Count, (1, 2)) self.assertIn(x.Root.Pages.Count, (1, 2))
if len(x.Root.Pages.Kids) == '1': if len(x.Root.Pages.Kids) == '1':
@ -646,10 +648,12 @@ def test_suite():
orig_img.close() orig_img.close()
except AttributeError: except AttributeError:
pass pass
if with_pdfrw: if engine == img2pdf.Engine.internal:
setattr(TestImg2Pdf, "test_%s_with_pdfrw" % test_name, handle) setattr(TestImg2Pdf, "test_%s_internal" % test_name, handle)
elif engine == img2pdf.Engine.pdfrw:
setattr(TestImg2Pdf, "test_%s_pdfrw" % test_name, handle)
else: else:
setattr(TestImg2Pdf, "test_%s_without_pdfrw" % test_name, handle) raise Exception("no such engine: %s"%engine)
return unittest.TestSuite(( return unittest.TestSuite((
unittest.makeSuite(TestImg2Pdf), unittest.makeSuite(TestImg2Pdf),