mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-05-13 01:13:54 +02:00
29abae0144
- update expected log messages - force `-ncu 3`, because otherwise the example images would not be deemed in need of enhancement
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import pytest
|
|
from PIL import Image
|
|
|
|
@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 / '2files/kant_aufklaerung_1784_0020.tif'
|
|
outfile = tmp_path / 'kant_aufklaerung_1784_0020.png'
|
|
run_eynollah_ok_and_check_logs(
|
|
'enhancement',
|
|
[
|
|
'-i', str(infile),
|
|
'-o', str(outfile.parent),
|
|
# force rescaling
|
|
'-ncu', 3,
|
|
] + options,
|
|
[
|
|
'Enhancement applied',
|
|
]
|
|
)
|
|
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',
|
|
[
|
|
'-di', str(resources_dir/ '2files'),
|
|
'-o', str(outdir),
|
|
# force rescaling
|
|
'-ncu', 3,
|
|
],
|
|
[
|
|
'Enhancement applied',
|
|
]
|
|
)
|
|
assert len(list(outdir.iterdir())) == 2
|