mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-13 13:36:15 +00:00
Harrison/fauna loader (#5864)
Co-authored-by: Shadid12 <Shadid12@users.noreply.github.com>
This commit is contained in:
41
tests/integration_tests/document_loaders/test_fauna.py
Normal file
41
tests/integration_tests/document_loaders/test_fauna.py
Normal file
@@ -0,0 +1,41 @@
|
||||
import unittest
|
||||
|
||||
from langchain.document_loaders.fauna import FaunaLoader
|
||||
|
||||
try:
|
||||
import fauna # noqa: F401
|
||||
|
||||
fauna_installed = True
|
||||
except ImportError:
|
||||
fauna_installed = False
|
||||
|
||||
|
||||
@unittest.skipIf(not fauna_installed, "fauna not installed")
|
||||
class TestFaunaLoader(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.fauna_secret = "<enter-valid-fauna-secret>"
|
||||
self.valid_fql_query = "Item.all()"
|
||||
self.valid_page_content_field = "text"
|
||||
self.valid_metadata_fields = ["valid_metadata_fields"]
|
||||
|
||||
def test_fauna_loader(self) -> None:
|
||||
"""Test Fauna loader."""
|
||||
loader = FaunaLoader(
|
||||
query=self.valid_fql_query,
|
||||
page_content_field=self.valid_page_content_field,
|
||||
secret=self.fauna_secret,
|
||||
metadata_fields=self.valid_metadata_fields,
|
||||
)
|
||||
docs = loader.load()
|
||||
|
||||
assert len(docs) > 0 # assuming the query returns at least one document
|
||||
for doc in docs:
|
||||
assert (
|
||||
doc.page_content != ""
|
||||
) # assuming that every document has page_content
|
||||
assert (
|
||||
"id" in doc.metadata and doc.metadata["id"] != ""
|
||||
) # assuming that every document has 'id'
|
||||
assert (
|
||||
"ts" in doc.metadata and doc.metadata["ts"] != ""
|
||||
) # assuming that every document has 'ts'
|
Reference in New Issue
Block a user