test_inmemory_similarity_search() — langchain Function Reference

Architecture documentation for the test_inmemory_similarity_search() function in test_in_memory.py from the langchain codebase.

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_inmemory_similarity_search() — langchain Function Reference

Dependency Diagram

graph TD
  a12d93e0_6fa2_f0d8_4faa_bba52a96d6cc["test_inmemory_similarity_search()"]
  a974d690_d5fa_ba65_671d_ce8278eefe7d["test_in_memory.py"]
  a12d93e0_6fa2_f0d8_4faa_bba52a96d6cc -->|defined in| a974d690_d5fa_ba65_671d_ce8278eefe7d
  style a12d93e0_6fa2_f0d8_4faa_bba52a96d6cc fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_in_memory.py lines 19–34

async def test_inmemory_similarity_search() -> None:
    """Test end to end similarity search."""
    store = await InMemoryVectorStore.afrom_texts(
        ["foo", "bar", "baz"], DeterministicFakeEmbedding(size=3)
    )

    # Check sync version
    output = store.similarity_search("foo", k=1)
    assert output == [_any_id_document(page_content="foo")]

    # Check async version
    output = await store.asimilarity_search("bar", k=2)
    assert output == [
        _any_id_document(page_content="bar"),
        _any_id_document(page_content="foo"),
    ]