mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-08-18 21:57:58 +00:00
25 lines
672 B
Python
25 lines
672 B
Python
|
import pytest
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
import unstructured.models.ocr.tesseract as tesseract
|
||
|
|
|
||
|
|
|
||
|
|
class MockTesseractAgent:
|
||
|
|
def __init__(self, languages):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def test_load_agent(monkeypatch):
|
||
|
|
monkeypatch.setattr(tesseract, "TesseractAgent", MockTesseractAgent)
|
||
|
|
|
||
|
|
with patch.object(tesseract, "is_pytesseract_available", return_value=True):
|
||
|
|
tesseract.load_agent()
|
||
|
|
|
||
|
|
assert isinstance(tesseract.ocr_agent, MockTesseractAgent)
|
||
|
|
|
||
|
|
|
||
|
|
def test_load_agent_raises_when_not_available():
|
||
|
|
with patch.object(tesseract, "is_pytesseract_available", return_value=False):
|
||
|
|
with pytest.raises(ImportError):
|
||
|
|
tesseract.load_agent()
|