refactor cli tests

This commit is contained in:
kba 2025-10-29 16:20:30 +01:00
parent ef999c8f0a
commit b6f82c72b9
15 changed files with 453 additions and 592 deletions

View file

@ -0,0 +1,57 @@
import pytest
from PIL import Image
from eynollah.cli import (
enhancement as enhancement_cli,
)
from ocrd_modelfactory import page_from_file
from ocrd_models.constants import NAMESPACES as NS
@pytest.mark.parametrize(
"options",
[
[], # defaults
["-sos"],
], ids=str)
def test_run_eynollah_enhancement_filename(
tmp_path,
resources_dir,
run_eynollah_ok_and_check_logs,
options,
):
infile = resources_dir / 'kant_aufklaerung_1784_0020.tif'
outfile = tmp_path.joinpath('kant_aufklaerung_1784_0020.png')
run_eynollah_ok_and_check_logs(
enhancement_cli,
[
'-i', str(infile),
'-o', str(outfile.parent),
] + options,
[
'Image was enhanced',
]
)
with Image.open(infile) as original_img:
original_size = original_img.size
with Image.open(outfile) as enhanced_img:
enhanced_size = enhanced_img.size
assert (original_size == enhanced_size) == ("-sos" in options)
def test_run_eynollah_enhancement_directory(
tmp_path,
resources_dir,
image_resources,
run_eynollah_ok_and_check_logs,
):
outdir = tmp_path
run_eynollah_ok_and_check_logs(
enhancement_cli,
[
'-di', str(resources_dir),
'-o', str(outdir),
],
[
f'Image {image_resources[0]} was enhanced',
f'Image {image_resources[1]} was enhanced',
]
)
assert len(list(outdir.iterdir())) == 2