22 lines
540 B
Python
22 lines
540 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
root_path = Path(__file__).parent.parent.parent
|
|
sys.path.append(str(root_path))
|
|
from pprint import pprint
|
|
|
|
test_files = {
|
|
"ocr_test.pdf": str(root_path / "tests" / "samples" / "ocr_test.pdf"),
|
|
}
|
|
|
|
def test_rapidocrpdfloader():
|
|
pdf_path = test_files["ocr_test.pdf"]
|
|
from document_loaders import RapidOCRPDFLoader
|
|
|
|
loader = RapidOCRPDFLoader(pdf_path)
|
|
docs = loader.load()
|
|
pprint(docs)
|
|
assert isinstance(docs, list) and len(docs) > 0 and isinstance(docs[0].page_content, str)
|
|
|
|
|