mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2026-02-08 05:13:50 -05:00
740f67091c
Technically OCROptions is more Pythonic but we have several pre-existing classes named OcrWhatever. Go with the local flow.
140 lines
4.4 KiB
Python
140 lines
4.4 KiB
Python
# SPDX-FileCopyrightText: 2025 James R. Barlow
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
|
|
"""Unit tests for OCR engine selection mechanism.
|
|
|
|
Tests verify that the --ocr-engine option works correctly and that
|
|
engine-specific options are available.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
class TestOcrEngineCliOption:
|
|
"""Test --ocr-engine CLI option."""
|
|
|
|
def test_ocr_engine_option_exists(self):
|
|
"""CLI should have --ocr-engine option."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
option_strings = []
|
|
for action in parser._actions:
|
|
option_strings.extend(action.option_strings)
|
|
|
|
assert '--ocr-engine' in option_strings
|
|
|
|
def test_ocr_engine_accepts_tesseract(self):
|
|
"""--ocr-engine should accept 'tesseract'."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
args = parser.parse_args(['--ocr-engine', 'tesseract', 'in.pdf', 'out.pdf'])
|
|
assert args.ocr_engine == 'tesseract'
|
|
|
|
def test_ocr_engine_accepts_auto(self):
|
|
"""--ocr-engine should accept 'auto'."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
args = parser.parse_args(['--ocr-engine', 'auto', 'in.pdf', 'out.pdf'])
|
|
assert args.ocr_engine == 'auto'
|
|
|
|
def test_ocr_engine_accepts_none(self):
|
|
"""--ocr-engine should accept 'none'."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
args = parser.parse_args(['--ocr-engine', 'none', 'in.pdf', 'out.pdf'])
|
|
assert args.ocr_engine == 'none'
|
|
|
|
def test_ocr_engine_default_is_auto(self):
|
|
"""--ocr-engine should default to 'auto'."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
args = parser.parse_args(['in.pdf', 'out.pdf'])
|
|
assert args.ocr_engine == 'auto'
|
|
|
|
def test_ocr_engine_rejects_invalid(self):
|
|
"""--ocr-engine should reject invalid values."""
|
|
from ocrmypdf.cli import get_parser
|
|
|
|
parser = get_parser()
|
|
|
|
with pytest.raises(SystemExit):
|
|
parser.parse_args(['--ocr-engine', 'invalid_engine', 'in.pdf', 'out.pdf'])
|
|
|
|
|
|
class TestOcrEngineOptionsModel:
|
|
"""Test OcrOptions has ocr_engine field."""
|
|
|
|
def test_ocr_options_has_ocr_engine_field(self):
|
|
"""OcrOptions should have ocr_engine field."""
|
|
from ocrmypdf._options import OcrOptions
|
|
|
|
# Check field exists in model
|
|
assert 'ocr_engine' in OcrOptions.model_fields
|
|
|
|
|
|
class TestOcrEnginePluginSelection:
|
|
"""Test that get_ocr_engine() hook selects correct engine based on options."""
|
|
|
|
def test_tesseract_selected_when_auto(self):
|
|
"""TesseractOcrEngine should be returned when ocr_engine='auto'."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from ocrmypdf.builtin_plugins import tesseract_ocr
|
|
from ocrmypdf.builtin_plugins.tesseract_ocr import TesseractOcrEngine
|
|
|
|
options = MagicMock()
|
|
options.ocr_engine = 'auto'
|
|
|
|
engine = tesseract_ocr.get_ocr_engine(options=options)
|
|
assert isinstance(engine, TesseractOcrEngine)
|
|
|
|
def test_tesseract_selected_when_tesseract(self):
|
|
"""TesseractOcrEngine should be returned when ocr_engine='tesseract'."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from ocrmypdf.builtin_plugins import tesseract_ocr
|
|
from ocrmypdf.builtin_plugins.tesseract_ocr import TesseractOcrEngine
|
|
|
|
options = MagicMock()
|
|
options.ocr_engine = 'tesseract'
|
|
|
|
engine = tesseract_ocr.get_ocr_engine(options=options)
|
|
assert isinstance(engine, TesseractOcrEngine)
|
|
|
|
def test_null_selected_when_none(self):
|
|
"""NullOcrEngine should be returned when ocr_engine='none'."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from ocrmypdf.builtin_plugins import null_ocr
|
|
from ocrmypdf.builtin_plugins.null_ocr import NullOcrEngine
|
|
|
|
options = MagicMock()
|
|
options.ocr_engine = 'none'
|
|
|
|
engine = null_ocr.get_ocr_engine(options=options)
|
|
assert isinstance(engine, NullOcrEngine)
|
|
|
|
def test_null_returns_none_when_auto(self):
|
|
"""null_ocr.get_ocr_engine() should return None when ocr_engine='auto'."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from ocrmypdf.builtin_plugins import null_ocr
|
|
|
|
options = MagicMock()
|
|
options.ocr_engine = 'auto'
|
|
|
|
engine = null_ocr.get_ocr_engine(options=options)
|
|
assert engine is None
|