2018-05-14 22:21:53 -07:00
|
|
|
# © 2018 James R. Barlow: github.com/jbarlow83
|
|
|
|
#
|
|
|
|
# This file is part of OCRmyPDF.
|
|
|
|
#
|
|
|
|
# OCRmyPDF is free software: you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# OCRmyPDF is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with OCRmyPDF. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2018-05-18 17:50:39 -07:00
|
|
|
from pathlib import Path
|
|
|
|
|
2018-05-14 22:21:53 -07:00
|
|
|
import pytest
|
2018-06-29 00:25:11 -07:00
|
|
|
import logging
|
|
|
|
|
|
|
|
from PIL import Image
|
2018-05-14 22:21:53 -07:00
|
|
|
|
2018-06-22 17:51:57 -07:00
|
|
|
from ocrmypdf import optimize as opt
|
2018-06-29 00:25:11 -07:00
|
|
|
from ocrmypdf.exec.ghostscript import rasterize_pdf
|
|
|
|
from ocrmypdf.helpers import fspath
|
2018-05-14 22:21:53 -07:00
|
|
|
|
|
|
|
|
2018-05-18 17:50:39 -07:00
|
|
|
@pytest.mark.parametrize('pdf', ['multipage.pdf', 'palette.pdf'])
|
|
|
|
def test_basic(resources, pdf, outpdf):
|
|
|
|
infile = resources / pdf
|
|
|
|
opt.main(infile, outpdf, level=3)
|
2018-05-14 22:21:53 -07:00
|
|
|
|
2018-05-18 17:50:39 -07:00
|
|
|
assert Path(outpdf).stat().st_size <= Path(infile).stat().st_size
|
2018-06-29 00:25:11 -07:00
|
|
|
|
|
|
|
|
|
|
|
def test_mono_not_inverted(resources, outdir):
|
|
|
|
infile = resources / '2400dpi.pdf'
|
|
|
|
opt.main(infile, outdir / 'out.pdf', level=3)
|
|
|
|
|
|
|
|
rasterize_pdf(
|
|
|
|
outdir / 'out.pdf', outdir / 'im.png',
|
|
|
|
xres=10, yres=10, raster_device='pnggray',
|
|
|
|
log=logging.getLogger(name='test_mono_flip')
|
|
|
|
)
|
|
|
|
|
|
|
|
im = Image.open(fspath(outdir / 'im.png'))
|
|
|
|
assert im.getpixel((0, 0)) == 255, "Expected white background"
|