James R. Barlow
1c8b763d53
The behavior of this test will ultimately depend on what version of img2pdf is installed, since after my patch it will be able to produce 1bpp images.
135 lines
3.7 KiB
Python
135 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
# © 2015 James R. Barlow: github.com/jbarlow83
|
|
|
|
from ocrmypdf import pageinfo
|
|
from reportlab.pdfgen.canvas import Canvas
|
|
from PIL import Image
|
|
from tempfile import NamedTemporaryFile
|
|
from contextlib import suppress
|
|
import os
|
|
import shutil
|
|
import pytest
|
|
import img2pdf
|
|
import pytest
|
|
import sys
|
|
|
|
|
|
if sys.version_info.major < 3:
|
|
print("Requires Python 3.4+")
|
|
sys.exit(1)
|
|
|
|
TESTS_ROOT = os.path.abspath(os.path.dirname(__file__))
|
|
SPOOF_PATH = os.path.join(TESTS_ROOT, 'spoof')
|
|
PROJECT_ROOT = os.path.dirname(TESTS_ROOT)
|
|
OCRMYPDF = os.path.join(PROJECT_ROOT, 'OCRmyPDF.sh')
|
|
TEST_RESOURCES = os.path.join(PROJECT_ROOT, 'tests', 'resources')
|
|
TEST_OUTPUT = os.environ.get(
|
|
'OCRMYPDF_TEST_OUTPUT',
|
|
default=os.path.join(PROJECT_ROOT, 'tests', 'output', 'pageinfo'))
|
|
|
|
|
|
def setup_module():
|
|
with suppress(FileNotFoundError):
|
|
shutil.rmtree(TEST_OUTPUT)
|
|
with suppress(FileExistsError):
|
|
os.makedirs(TEST_OUTPUT)
|
|
|
|
|
|
def _make_input(input_basename):
|
|
return os.path.join(TEST_RESOURCES, input_basename)
|
|
|
|
|
|
def _make_output(output_basename):
|
|
return os.path.join(TEST_OUTPUT, output_basename)
|
|
|
|
|
|
def test_single_page_text():
|
|
filename = os.path.join(TEST_OUTPUT, 'text.pdf')
|
|
pdf = Canvas(filename, pagesize=(8*72, 6*72))
|
|
text = pdf.beginText()
|
|
text.setFont('Helvetica', 12)
|
|
text.setTextOrigin(1*72, 3*72)
|
|
text.textLine("Methink'st thou art a general offence and every"
|
|
" man should beat thee.")
|
|
pdf.drawText(text)
|
|
pdf.showPage()
|
|
pdf.save()
|
|
|
|
pdfinfo = pageinfo.pdf_get_all_pageinfo(filename)
|
|
|
|
assert len(pdfinfo) == 1
|
|
page = pdfinfo[0]
|
|
|
|
assert page['has_text']
|
|
assert len(page['images']) == 0
|
|
|
|
|
|
def test_single_page_image():
|
|
filename = os.path.join(TEST_OUTPUT, 'image-mono.pdf')
|
|
|
|
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')
|
|
|
|
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)
|
|
|
|
pdfinfo = pageinfo.pdf_get_all_pageinfo(filename)
|
|
|
|
assert len(pdfinfo) == 1
|
|
page = pdfinfo[0]
|
|
|
|
assert not page['has_text']
|
|
assert len(page['images']) == 1
|
|
|
|
pdfimage = page['images'][0]
|
|
assert pdfimage['width'] == 8
|
|
assert pdfimage['color'] == 'gray'
|
|
|
|
# DPI in a 1"x1" is the image width
|
|
assert abs(pdfimage['dpi_w'] - 8) < 1e-5
|
|
assert abs(pdfimage['dpi_h'] - 8) < 1e-5
|
|
|
|
|
|
def test_single_page_inline_image():
|
|
filename = os.path.join(TEST_OUTPUT, 'image-mono-inline.pdf')
|
|
pdf = Canvas(filename, pagesize=(8*72, 6*72))
|
|
with NamedTemporaryFile() 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')
|
|
# Draw image in a 72x72 pt or 1"x1" area
|
|
pdf.drawInlineImage(im_tmp.name, 0, 0, width=72, height=72)
|
|
pdf.showPage()
|
|
pdf.save()
|
|
|
|
pdfinfo = pageinfo.pdf_get_all_pageinfo(filename)
|
|
print(pdfinfo)
|
|
pdfimage = pdfinfo[0]['images'][0]
|
|
assert (pdfimage['dpi_w'] - 8) < 1e-5
|
|
assert pdfimage['color'] != '-'
|
|
assert pdfimage['width'] == 8
|
|
|
|
|
|
def test_jpeg():
|
|
filename = _make_input('c02-22.pdf')
|
|
|
|
pdfinfo = pageinfo.pdf_get_all_pageinfo(filename)
|
|
|
|
pdfimage = pdfinfo[0]['images'][0]
|
|
assert pdfimage['enc'] == 'jpeg'
|
|
assert (pdfimage['dpi_w'] - 150) < 1e-5
|
|
|