James R. Barlow
2c24f67deb
Tesseract 3.05.01 backported the textonly_pdf=1 which allows the use of this superior PDF renderer prior to 4.00 alpha. This means that the tess4 name is no longer accurate, so call it a sandwich because of its merge-preserve characteristic. Preserve the tess4 name. Fix the documentation and tests to reflect this. Make it the default, because it’s better. It does not have the issues the “tesseract” renderer does prior to Tess 3.05.00 with rendering PDFs that Ghostscript corrupts, and it produces better output without re-rastering. Deprecate some old stuff to avoid the test suite growing obscenely large.
131 lines
3.6 KiB
Python
131 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
# © 2017 James R. Barlow: github.com/jbarlow83
|
|
|
|
import pytest
|
|
from ocrmypdf.exceptions import ExitCode
|
|
from ocrmypdf.exec import tesseract
|
|
from ocrmypdf import pdfinfo
|
|
import sys
|
|
import os
|
|
import PyPDF2 as pypdf
|
|
from contextlib import contextmanager
|
|
|
|
|
|
spoof = pytest.helpers.spoof
|
|
|
|
|
|
@pytest.fixture
|
|
def ensure_tess4():
|
|
if tesseract.v4():
|
|
# Either "tesseract" on $PATH is already v4, or
|
|
# OCRMYPDF_TESSERACT is tess4 already
|
|
return os.environ.copy()
|
|
|
|
if os.environ.get('OCRMYPDF_TESS4'):
|
|
# OCRMYPDF_TESS4 is a hint environment variable that tells us to look
|
|
# somewhere special for tess4 if and only if we need it. This allows
|
|
# setting OCRMYPDF_TESS4 to test tess4 and PATH to point to tess3
|
|
# on a system with both installed.
|
|
env = os.environ.copy()
|
|
env['OCRMYPDF_TESSERACT'] = env['OCRMYPDF_TESS4']
|
|
return env
|
|
|
|
raise EnvironmentError("Can't find Tesseract 4")
|
|
|
|
|
|
@contextmanager
|
|
def modified_os_environ(env):
|
|
old_env = os.environ.copy()
|
|
os.environ = env
|
|
yield
|
|
os.environ = old_env
|
|
|
|
|
|
def tess4_available():
|
|
"""Check if a tesseract 4 binary is available, even if it's not the
|
|
official "tesseract" on PATH
|
|
|
|
"""
|
|
try:
|
|
# ensure_tess4 locates the tess4 binary we are going to check
|
|
env = ensure_tess4()
|
|
with modified_os_environ(env):
|
|
# Now jump into this environment and make sure it really is Tess4
|
|
return tesseract.v4() and tesseract.has_textonly_pdf()
|
|
except EnvironmentError:
|
|
pass
|
|
|
|
return False
|
|
|
|
# Skip all tests in this file if not tesseract 4
|
|
pytestmark = pytest.mark.skipif(
|
|
not tess4_available(),
|
|
reason="tesseract 4.0 with textonly_pdf feature required")
|
|
|
|
check_ocrmypdf = pytest.helpers.check_ocrmypdf
|
|
run_ocrmypdf = pytest.helpers.run_ocrmypdf
|
|
spoof = pytest.helpers.spoof
|
|
|
|
|
|
def test_textonly_pdf(ensure_tess4, resources, outdir):
|
|
check_ocrmypdf(
|
|
resources / 'linn.pdf',
|
|
outdir / 'linn_textonly.pdf', '--pdf-renderer', 'tess4',
|
|
'--sidecar', 'foo',
|
|
env=ensure_tess4)
|
|
|
|
|
|
def test_pagesize_consistency_tess4(ensure_tess4, resources, outpdf):
|
|
from math import isclose
|
|
|
|
infile = resources / 'linn.pdf'
|
|
|
|
before_dims = pytest.helpers.first_page_dimensions(infile)
|
|
|
|
check_ocrmypdf(
|
|
infile,
|
|
outpdf, '--pdf-renderer', 'tess4',
|
|
'--clean', '--deskew', '--remove-background', '--clean-final',
|
|
env=ensure_tess4)
|
|
|
|
after_dims = pytest.helpers.first_page_dimensions(outpdf)
|
|
|
|
assert isclose(before_dims[0], after_dims[0])
|
|
assert isclose(before_dims[1], after_dims[1])
|
|
|
|
|
|
@pytest.mark.parametrize('basename', ['graph_ocred.pdf', 'cardinal.pdf'])
|
|
def test_skip_pages_does_not_replicate(
|
|
ensure_tess4, resources, basename, outdir):
|
|
infile = resources / basename
|
|
outpdf = outdir / basename
|
|
|
|
check_ocrmypdf(
|
|
infile,
|
|
outpdf, '--pdf-renderer', 'tess4', '--force-ocr',
|
|
'--tesseract-timeout', '0',
|
|
env=ensure_tess4
|
|
)
|
|
|
|
info_in = pdfinfo.PdfInfo(infile)
|
|
|
|
info = pdfinfo.PdfInfo(outpdf)
|
|
for page in info:
|
|
assert len(page.images) == 1, "skipped page was replicated"
|
|
|
|
for n in range(len(info_in)):
|
|
assert info[n].width_inches == info_in[n].width_inches
|
|
|
|
|
|
def test_content_preservation(ensure_tess4, resources, outpdf):
|
|
infile = resources / 'masks.pdf'
|
|
|
|
check_ocrmypdf(
|
|
infile,
|
|
outpdf, '--pdf-renderer', 'tess4', '--tesseract-timeout', '0',
|
|
env=ensure_tess4
|
|
)
|
|
|
|
info = pdfinfo.PdfInfo(outpdf)
|
|
page = info[0]
|
|
assert len(page.images) > 1, "masks were rasterized" |