dangerzone/tests/test_ocr.py
2024-10-09 18:57:33 +03:00

16 lines
670 B
Python

from dangerzone.isolation_provider.dummy import Dummy
from dangerzone.logic import DangerzoneCore
from dangerzone.util import get_tessdata_dir
def test_ocr_ommisions() -> None:
# Grab the languages that are available in the Tesseract data dir.
suffix_len = len(".traineddata")
available_langs = {f.name[:-suffix_len] for f in get_tessdata_dir().iterdir()}
# Grab the languages that Dangerzone offers to the user through the GUI/CLI.
offered_langs = set(DangerzoneCore(Dummy()).ocr_languages.values())
# Ensure that both the available languages and the ones we offer to the user are the
# same.
assert available_langs == offered_langs