eynollah/tests/conftest.py
2025-11-06 16:33:39 +01:00

37 lines
806 B
Python

from glob import glob
import os
import pytest
from pathlib import Path
@pytest.fixture()
def tests_dir():
return Path(__file__).parent.resolve()
@pytest.fixture()
def model_dir(tests_dir):
return os.environ.get('EYNOLLAH_MODELS_DIR', str(tests_dir.joinpath('..').resolve()))
@pytest.fixture()
def resources_dir(tests_dir):
return tests_dir / 'resources/2files'
@pytest.fixture()
def image_resources(resources_dir):
return [Path(x) for x in glob(str(resources_dir / '*.tif'))]
@pytest.fixture()
def eynollah_log_filter():
return lambda logrec: logrec.name.startswith('eynollah')
@pytest.fixture
def eynollah_subcommands():
return [
'binarization',
'layout',
'ocr',
'enhancement',
'machine-based-reading-order',
'models',
]