mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2025-08-15 20:21:21 +00:00
127 lines
3.4 KiB
Python
127 lines
3.4 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.mkdir(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() 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)
|
|
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'
|
|
|
|
# While unexpected, this is correct
|
|
# PDF spec says /FlateDecode image must have /BitsPerComponent 8
|
|
# So mono images get upgraded to 8-bit
|
|
assert pdfimage['bpc'] == 8
|
|
|
|
# DPI in a 1"x1" is the image width
|
|
assert pdfimage['dpi_w'] == 8
|
|
assert pdfimage['dpi_h'] == 8
|
|
|
|
|
|
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()
|
|
|
|
with pytest.raises(NotImplementedError):
|
|
pageinfo.pdf_get_all_pageinfo(filename)
|
|
|
|
|
|
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'
|
|
|