mirror of
https://github.com/qurator-spk/dinglehopper.git
synced 2025-11-04 11:24:17 +01:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import os
|
|
import shutil
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from click.testing import CliRunner
|
|
from .util import working_directory
|
|
|
|
|
|
from ..ocrd_cli import ocrd_dinglehopper
|
|
|
|
data_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data')
|
|
|
|
|
|
def test_ocrd_cli(tmp_path):
|
|
"""Test OCR-D interface"""
|
|
|
|
# Copy test workspace
|
|
test_workspace_dir_source = Path(data_dir) / 'actevedef_718448162'
|
|
test_workspace_dir = tmp_path / 'test_ocrd_cli'
|
|
shutil.copytree(str(test_workspace_dir_source), str(test_workspace_dir))
|
|
|
|
# Run through the OCR-D interface
|
|
with working_directory(str(test_workspace_dir)):
|
|
runner = CliRunner()
|
|
args = [
|
|
'-m', 'mets.xml',
|
|
'-I', 'OCR-D-GT-PAGE,OCR-D-OCR-CALAMARI',
|
|
'-O', 'OCR-D-OCR-CALAMARI-EVAL'
|
|
]
|
|
sys.argv[1:] = args # XXX Hack to satisfy ocrd_cli_wrap_processor() check for arguments
|
|
result = runner.invoke(ocrd_dinglehopper, args)
|
|
assert result.exit_code == 0
|
|
result_json = list((test_workspace_dir / 'OCR-D-OCR-CALAMARI-EVAL').glob('*.json'))
|
|
assert json.load(open(str(result_json[0])))['cer'] < 0.03
|