test_similarity_search() — langchain Function Reference

Architecture documentation for the test_similarity_search() function in test_vectorstores.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_similarity_search() — langchain Function Reference

Dependency Diagram

graph TD
  88d7ab42_4e08_d016_c60e_143512ada144["test_similarity_search()"]
  1e8a5ebb_88eb_33d3_2eef_6d4dfedf6943["test_vectorstores.py"]
  88d7ab42_4e08_d016_c60e_143512ada144 -->|defined in| 1e8a5ebb_88eb_33d3_2eef_6d4dfedf6943
  style 88d7ab42_4e08_d016_c60e_143512ada144 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/chroma/tests/unit_tests/test_vectorstores.py lines 18–30

def test_similarity_search() -> None:
    """Test similarity search by Chroma."""
    texts = ["foo", "bar", "baz"]
    metadatas = [{"page": str(i)} for i in range(len(texts))]
    docsearch = Chroma.from_texts(
        collection_name="test_collection",
        texts=texts,
        embedding=FakeEmbeddings(size=10),
        metadatas=metadatas,
    )
    output = docsearch.similarity_search("foo", k=1)
    docsearch.delete_collection()
    assert len(output) == 1