mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2025-08-15 04:01:40 +00:00
81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
# © 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/>.
|
|
|
|
|
|
import os
|
|
import shutil
|
|
import sys
|
|
from pickle import dumps, loads
|
|
|
|
import pytest
|
|
from PIL import Image, ImageChops
|
|
|
|
import ocrmypdf.leptonica as lept
|
|
|
|
|
|
def test_colormap_backgroundnorm(resources):
|
|
# Issue #262 - unclear how to reproduce exactly, so just ensure leptonica
|
|
# can handle that case
|
|
pix = lept.Pix.open(resources / 'baiona_colormapped.png')
|
|
pix.background_norm()
|
|
|
|
|
|
@pytest.fixture
|
|
def crom_pix(resources):
|
|
pix = lept.Pix.open(resources / 'crom.png')
|
|
im = Image.open(resources / 'crom.png')
|
|
return pix, im
|
|
|
|
|
|
def test_pix_basic(crom_pix):
|
|
pix, im = crom_pix
|
|
|
|
assert pix.width == im.width
|
|
assert pix.height == im.height
|
|
assert pix.mode == im.mode
|
|
|
|
|
|
def test_pil_conversion(crom_pix):
|
|
pix, im = crom_pix
|
|
|
|
# Check for pixel perfect
|
|
assert ImageChops.difference(pix.topil(), im).getbbox() is None
|
|
|
|
|
|
def test_pix_otsu(crom_pix):
|
|
pix, _ = crom_pix
|
|
im1bpp = pix.otsu_adaptive_threshold()
|
|
assert im1bpp.mode == '1'
|
|
|
|
|
|
def test_crop(resources):
|
|
pix = lept.Pix.open(resources / 'linn.png')
|
|
foreground = pix.crop_to_foreground()
|
|
assert foreground.width < pix.width
|
|
|
|
|
|
def test_clean_bg(crom_pix):
|
|
pix, _ = crom_pix
|
|
imbg = pix.clean_background_to_white()
|
|
|
|
|
|
def test_pickle(crom_pix):
|
|
pix, _ = crom_pix
|
|
pickled = dumps(pix)
|
|
pix2 = loads(pickled)
|
|
assert pix.mode == pix2.mode
|