mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2026-02-08 05:13:50 -05:00
0c3745a1a4
Introduce --ocr-engine option to select between OCR engines: - 'auto' (default): Uses Tesseract - 'tesseract': Explicit Tesseract selection - 'none': Skip OCR entirely (for PDF processing only) Key changes: - Extend OcrEngine ABC with generate_ocr() and supports_generate_ocr() for direct OcrElement tree output (bypasses hOCR) - Add get_ocr_engine(options) hook parameter for engine selection - Implement NullOcrEngine for --ocr-engine none - Export OcrElement, OcrClass, BoundingBox from ocrmypdf package - Add ocr_tree support to grafting pipeline This prepares the foundation for pluggable OCR engines while maintaining full backward compatibility with existing Tesseract-based workflows.
132 lines
3.9 KiB
Python
132 lines
3.9 KiB
Python
# SPDX-FileCopyrightText: 2025 James R. Barlow
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
|
|
"""Unit tests for OcrEngine interface extensions.
|
|
|
|
These tests verify that the OcrEngine ABC has the new generate_ocr() method
|
|
and that OcrElement classes are exported from the public API.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from ocrmypdf.pluginspec import OcrEngine
|
|
|
|
|
|
class TestOcrEngineInterface:
|
|
"""Test that OcrEngine ABC has required methods."""
|
|
|
|
def test_generate_ocr_method_exists(self):
|
|
"""OcrEngine must have generate_ocr() method signature."""
|
|
assert hasattr(OcrEngine, 'generate_ocr')
|
|
|
|
def test_supports_generate_ocr_method_exists(self):
|
|
"""OcrEngine must have supports_generate_ocr() method."""
|
|
assert hasattr(OcrEngine, 'supports_generate_ocr')
|
|
|
|
def test_supports_generate_ocr_default_false(self):
|
|
"""Default supports_generate_ocr() should return False."""
|
|
from ocrmypdf.pluginspec import OrientationConfidence
|
|
|
|
# Create a minimal concrete implementation
|
|
class MinimalEngine(OcrEngine):
|
|
@staticmethod
|
|
def version():
|
|
return "1.0"
|
|
|
|
@staticmethod
|
|
def creator_tag(options):
|
|
return "test"
|
|
|
|
def __str__(self):
|
|
return "test"
|
|
|
|
@staticmethod
|
|
def languages(options):
|
|
return set()
|
|
|
|
@staticmethod
|
|
def get_orientation(input_file, options):
|
|
return OrientationConfidence(0, 0.0)
|
|
|
|
@staticmethod
|
|
def get_deskew(input_file, options):
|
|
return 0.0
|
|
|
|
@staticmethod
|
|
def generate_hocr(input_file, output_hocr, output_text, options):
|
|
pass
|
|
|
|
@staticmethod
|
|
def generate_pdf(input_file, output_pdf, output_text, options):
|
|
pass
|
|
|
|
engine = MinimalEngine()
|
|
assert engine.supports_generate_ocr() is False
|
|
|
|
def test_generate_ocr_raises_not_implemented_by_default(self):
|
|
"""Default generate_ocr() should raise NotImplementedError."""
|
|
from ocrmypdf.pluginspec import OrientationConfidence
|
|
|
|
class MinimalEngine(OcrEngine):
|
|
@staticmethod
|
|
def version():
|
|
return "1.0"
|
|
|
|
@staticmethod
|
|
def creator_tag(options):
|
|
return "test"
|
|
|
|
def __str__(self):
|
|
return "test"
|
|
|
|
@staticmethod
|
|
def languages(options):
|
|
return set()
|
|
|
|
@staticmethod
|
|
def get_orientation(input_file, options):
|
|
return OrientationConfidence(0, 0.0)
|
|
|
|
@staticmethod
|
|
def get_deskew(input_file, options):
|
|
return 0.0
|
|
|
|
@staticmethod
|
|
def generate_hocr(input_file, output_hocr, output_text, options):
|
|
pass
|
|
|
|
@staticmethod
|
|
def generate_pdf(input_file, output_pdf, output_text, options):
|
|
pass
|
|
|
|
engine = MinimalEngine()
|
|
with pytest.raises(NotImplementedError):
|
|
engine.generate_ocr(Path("test.png"), MagicMock(), 0)
|
|
|
|
|
|
class TestOcrElementExport:
|
|
"""Test that OcrElement is exported from public API."""
|
|
|
|
def test_ocrelement_importable_from_ocrmypdf(self):
|
|
"""OcrElement should be importable from ocrmypdf package."""
|
|
from ocrmypdf import OcrElement
|
|
|
|
assert OcrElement is not None
|
|
|
|
def test_ocrclass_importable_from_ocrmypdf(self):
|
|
"""OcrClass should be importable from ocrmypdf package."""
|
|
from ocrmypdf import OcrClass
|
|
|
|
assert OcrClass is not None
|
|
|
|
def test_boundingbox_importable_from_ocrmypdf(self):
|
|
"""BoundingBox should be importable from ocrmypdf package."""
|
|
from ocrmypdf import BoundingBox
|
|
|
|
assert BoundingBox is not None
|