mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
e5f6f0ffc4
## Description Support search params in GoogleSearchApiWrapper's result call, for the extra filtering on search, to support extra query parameters that google cse provides: https://developers.google.com/custom-search/v1/reference/rest/v1/cse/list?hl=ko ## Issue #6810
31 lines
998 B
Python
31 lines
998 B
Python
"""Integration test for Google Search API Wrapper."""
|
|
from langchain.utilities.google_search import GoogleSearchAPIWrapper
|
|
|
|
|
|
def test_call() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = GoogleSearchAPIWrapper()
|
|
output = search.run("What was Obama's first name?")
|
|
assert "Barack Hussein Obama II" in output
|
|
|
|
|
|
def test_no_result_call() -> None:
|
|
"""Test that call gives no result."""
|
|
search = GoogleSearchAPIWrapper()
|
|
output = search.run(
|
|
"NORESULTCALL_NORESULTCALL_NORESULTCALL_NORESULTCALL_NORESULTCALL_NORESULTCALL"
|
|
)
|
|
print(type(output))
|
|
assert "No good Google Search Result was found" == output
|
|
|
|
|
|
def test_result_with_params_call() -> None:
|
|
"""Test that call gives the correct answer with extra params."""
|
|
search = GoogleSearchAPIWrapper()
|
|
output = search.results(
|
|
query="What was Obama's first name?",
|
|
num_results=5,
|
|
search_params={"cr": "us", "safe": "active"},
|
|
)
|
|
assert len(output)
|