mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
9ec60ad832
All credit to @UmerHA, made a couple small changes --------- Co-authored-by: UmerHA <40663591+UmerHA@users.noreply.github.com>
25 lines
899 B
Python
25 lines
899 B
Python
"""Test Azure Cognitive Search wrapper."""
|
|
import pytest
|
|
|
|
from langchain.retrievers.azure_cognitive_search import AzureCognitiveSearchRetriever
|
|
from langchain.schema import Document
|
|
|
|
|
|
def test_azure_cognitive_search_get_relevant_documents() -> None:
|
|
"""Test valid call to Azure Cognitive Search."""
|
|
retriever = AzureCognitiveSearchRetriever()
|
|
documents = retriever.get_relevant_documents("what is langchain")
|
|
for doc in documents:
|
|
assert isinstance(doc, Document)
|
|
assert doc.page_content
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_azure_cognitive_search_aget_relevant_documents() -> None:
|
|
"""Test valid async call to Azure Cognitive Search."""
|
|
retriever = AzureCognitiveSearchRetriever()
|
|
documents = await retriever.aget_relevant_documents("what is langchain")
|
|
for doc in documents:
|
|
assert isinstance(doc, Document)
|
|
assert doc.page_content
|