mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
25fbe356b4
This PR upgrades community to a recent version of mypy. It inserts type: ignore on all existing failures.
10 lines
448 B
Python
10 lines
448 B
Python
"""Integration test for Dataherald API Wrapper."""
|
|
from langchain_community.utilities.dataherald import DataheraldAPIWrapper
|
|
|
|
|
|
def test_call() -> None:
|
|
"""Test that call gives the correct answer."""
|
|
search = DataheraldAPIWrapper(db_connection_id="65fb766367dd22c99ce1a12d") # type: ignore[call-arg]
|
|
output = search.run("How many employees are in the company?")
|
|
assert "Answer: SELECT \n COUNT(*) FROM \n employees" in output
|