langchain/libs/community/tests/integration_tests/utilities/test_golden_query_api.py

"""Integration test for Golden API Wrapper."""
import json

from langchain_community.utilities.golden_query import GoldenQueryAPIWrapper


def test_call() -> None:
    """Test that call gives the correct answer."""
    search = GoldenQueryAPIWrapper()
    output = json.loads(search.run("companies in nanotech"))
    assert len(output.get("results", [])) > 0