From 0dc96442d8ed0309d5baffd5fcd4215b36f1010e Mon Sep 17 00:00:00 2001 From: "James R. Barlow" Date: Sat, 6 Feb 2016 23:41:32 -0800 Subject: [PATCH] Fix img2pdf usage in test case (to make Travis CI happy again) --- tests/test_pageinfo.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/test_pageinfo.py b/tests/test_pageinfo.py index dc994001..8968c322 100644 --- a/tests/test_pageinfo.py +++ b/tests/test_pageinfo.py @@ -67,13 +67,21 @@ def test_single_page_text(): def test_single_page_image(): filename = os.path.join(TEST_OUTPUT, 'image-mono.pdf') - with NamedTemporaryFile() as im_tmp: + with NamedTemporaryFile(mode='wb+', suffix='.png') as im_tmp: im = Image.new('1', (8, 8), 0) for n in range(8): im.putpixel((n, n), 1) im.save(im_tmp.name, format='PNG') - pdf_bytes = img2pdf.convert([im_tmp.name], dpi=8) + imgsize = ((img2pdf.ImgSize.dpi, 8), (img2pdf.ImgSize.dpi, 8)) + layout_fun = img2pdf.get_layout_fun(None, imgsize, None, None, None) + + im_tmp.seek(0) + im_bytes = im_tmp.read() + pdf_bytes = img2pdf.convert( + im_bytes, producer="img2pdf", with_pdfrw=False, + layout_fun=layout_fun) + with open(filename, 'wb') as pdf: pdf.write(pdf_bytes)