Commit e1073665 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add cache clear

parent 822b8820
Loading
Loading
Loading
Loading
+12 −0
Original line number Diff line number Diff line
@@ -2,9 +2,21 @@ import pytest
from PIL import Image

from imgutils.ocr import detect_text_with_ocr, list_det_models, list_rec_models, ocr
from imgutils.ocr.detect import _open_ocr_detection_model
from imgutils.ocr.recognize import _open_ocr_recognition_dictionary, _open_ocr_recognition_model
from test.testings import get_testfile


@pytest.fixture(autouse=True, scope='module')
def _clear_cache():
    try:
        yield
    finally:
        _open_ocr_detection_model.cache_clear()
        _open_ocr_recognition_model.cache_clear()
        _open_ocr_recognition_dictionary.cache_clear()


@pytest.fixture()
def ocr_img_plot():
    yield get_testfile('ocr', 'plot.png')