Harrison/serper api bug (#4902)

Co-authored-by: Jerry Luan <xmaswillyou@gmail.com>
This commit is contained in:
Harrison Chase
2023-05-17 21:40:39 -07:00
committed by GitHub
parent c998569c8f
commit 9e2227ba11
2 changed files with 24 additions and 7 deletions

View File

@@ -4,13 +4,20 @@ import pytest
from langchain.utilities.google_serper import GoogleSerperAPIWrapper
def test_call() -> None:
"""Test that call gives the correct answer."""
def test_search_call() -> None:
"""Test that call gives the correct answer from search."""
search = GoogleSerperAPIWrapper()
output = search.run("What was Obama's first name?")
assert "Barack Hussein Obama II" in output
def test_news_call() -> None:
"""Test that call gives the correct answer from news search."""
search = GoogleSerperAPIWrapper(type="news")
output = search.run("What's new with stock market?").lower()
assert "stock" in output or "market" in output
async def test_results() -> None:
"""Test that call gives the correct answer."""
search = GoogleSerperAPIWrapper()