weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix unit test name

This commit is contained in:
imClumsyPanda 2023-09-05 21:07:54 +08:00
parent 082f429809
commit 0ea18314f2
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/document_loader/test_imgloader.py
View File

@ -6,14 +6,14 @@ sys.path.append(str(root_path))
from pprint import pprint from pprint import pprint
test_files = { test_files = {
"ocr_test.pdf": str(root_path / "tests" / "samples" / "ocr_test.pdf"), "ocr_test.jpg": str(root_path / "tests" / "samples" / "ocr_test.jpg"),
} }
def test_rapidocrpdfloader(): def test_rapidocrloader():
pdf_path = test_files["ocr_test.pdf"] img_path = test_files["ocr_test.jpg"]
from document_loaders import RapidOCRPDFLoader from document_loaders import RapidOCRLoader
loader = RapidOCRPDFLoader(pdf_path) loader = RapidOCRLoader(img_path)
docs = loader.load() docs = loader.load()
pprint(docs) pprint(docs)
assert isinstance(docs, list) and len(docs) > 0 and isinstance(docs[0].page_content, str) assert isinstance(docs, list) and len(docs) > 0 and isinstance(docs[0].page_content, str)

10
tests/document_loader/test_pdfloader.py
View File

@ -6,14 +6,14 @@ sys.path.append(str(root_path))
from pprint import pprint from pprint import pprint
test_files = { test_files = {
"ocr_test.jpg": str(root_path / "tests" / "samples" / "ocr_test.jpg"), "ocr_test.pdf": str(root_path / "tests" / "samples" / "ocr_test.pdf"),
} }
def test_rapidocrloader(): def test_rapidocrpdfloader():
img_path = test_files["ocr_test.jpg"] pdf_path = test_files["ocr_test.pdf"]
from document_loaders import RapidOCRLoader from document_loaders import RapidOCRPDFLoader
loader = RapidOCRLoader(img_path) loader = RapidOCRPDFLoader(pdf_path)
docs = loader.load() docs = loader.load()
pprint(docs) pprint(docs)
assert isinstance(docs, list) and len(docs) > 0 and isinstance(docs[0].page_content, str) assert isinstance(docs, list) and len(docs) > 0 and isinstance(docs[0].page_content, str)