17 lines
408 B
Python
17 lines
408 B
Python
import json
|
|
from pathlib import Path
|
|
|
|
from manga_ocr import MangaOcr
|
|
|
|
TEST_DATA_ROOT = Path(__file__).parent / 'data'
|
|
|
|
|
|
def test_ocr():
|
|
mocr = MangaOcr()
|
|
|
|
expected_results = json.loads((TEST_DATA_ROOT / 'expected_results.json').read_text(encoding='utf-8'))
|
|
|
|
for item in expected_results:
|
|
result = mocr(TEST_DATA_ROOT / 'images' / item['filename'])
|
|
assert result == item['result']
|