test_search_tool_run() — langchain Function Reference

Architecture documentation for the test_search_tool_run() function in test_tools.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_search_tool_run() — langchain Function Reference

Dependency Diagram

graph TD
  b14fa70b_45e5_4e4a_d911_5cfcdd78a41c["test_search_tool_run()"]
  d59061e5_8103_862f_4af9_770edb6e5f3a["test_tools.py"]
  b14fa70b_45e5_4e4a_d911_5cfcdd78a41c -->|defined in| d59061e5_8103_862f_4af9_770edb6e5f3a
  style b14fa70b_45e5_4e4a_d911_5cfcdd78a41c fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/perplexity/tests/unit_tests/test_tools.py lines 8–35

def test_search_tool_run(mocker: MockerFixture) -> None:
    tool = PerplexitySearchResults(pplx_api_key="test")

    mock_result = MagicMock()
    mock_result.title = "Test Title"
    mock_result.url = "http://test.com"
    mock_result.snippet = "Test snippet"
    mock_result.date = "2023-01-01"
    mock_result.last_updated = "2023-01-02"

    mock_response = MagicMock()
    mock_response.results = [mock_result]

    mock_create = MagicMock(return_value=mock_response)
    mocker.patch.object(tool.client.search, "create", mock_create)

    result = tool.invoke("query")

    # result should be a list of dicts (converted by tool) or str if string output
    # By default, tool.invoke returns the output of _run.
    assert isinstance(result, list)
    assert len(result) == 1
    assert result[0]["title"] == "Test Title"

    mock_create.assert_called_once_with(
        query="query",
        max_results=10,
    )