mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
b1446bea5f
* implemented arun, results, and aresults. Reuses aiosession if available. * helper tools GoogleSerperRun and GoogleSerperResults * support for Google Images, Places and News (examples given) and filtering based on time (e.g. past hour) * updated docs
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""Integration test for Serper.dev's Google Search API Wrapper."""
|
|
import pytest
|
|
|
|
from langchain.utilities.google_serper import GoogleSerperAPIWrapper
|
|
|
|
|
|
def test_call() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = GoogleSerperAPIWrapper()
|
|
output = search.run("What was Obama's first name?")
|
|
assert "Barack Hussein Obama II" in output
|
|
|
|
|
|
async def test_results() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = GoogleSerperAPIWrapper()
|
|
output = search.results("What was Obama's first name?")
|
|
assert "Barack Hussein Obama II" in output["answerBox"]["answer"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_call() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = GoogleSerperAPIWrapper()
|
|
output = await search.arun("What was Obama's first name?")
|
|
assert "Barack Hussein Obama II" in output
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_results() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = GoogleSerperAPIWrapper()
|
|
output = await search.aresults("What was Obama's first name?")
|
|
assert "Barack Hussein Obama II" in output["answerBox"]["answer"]
|