File size: 469 Bytes
e8013a0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import pytest
from scr.utils import extract_text_from_pdf, ModelError
from unittest.mock import patch
def test_extract_text_from_pdf():
with patch("scr.utils.extract_text") as mock_extract:
mock_extract.return_value = "Dummy text from PDF"
result = extract_text_from_pdf("dummy_path.pdf")
assert result == "Dummy text from PDF"
def test_model_error():
with pytest.raises(ModelError):
raise ModelError("This is a test error")
|