test_chroma_with_ids() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_chroma_with_ids() — langchain Function Reference

Dependency Diagram

graph TD
  72c7c0b7_b46a_0c5b_e137_352f37be1913["test_chroma_with_ids()"]
  289e5a4f_8e2e_ddb6_af2e_804f91dabdb8["test_vectorstores.py"]
  72c7c0b7_b46a_0c5b_e137_352f37be1913 -->|defined in| 289e5a4f_8e2e_ddb6_af2e_804f91dabdb8
  style 72c7c0b7_b46a_0c5b_e137_352f37be1913 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/chroma/tests/integration_tests/test_vectorstores.py lines 73–88

def test_chroma_with_ids() -> None:
    """Test end to end construction and search."""
    texts = ["foo", "bar", "baz"]
    ids = [f"id_{i}" for i in range(len(texts))]
    docsearch = Chroma.from_texts(
        collection_name="test_collection",
        texts=texts,
        embedding=FakeEmbeddings(),
        ids=ids,
    )
    output = docsearch.similarity_search("foo", k=1)

    docsearch.delete_collection()
    assert len(output) == 1
    assert output[0].page_content == "foo"
    assert output[0].id == "id_0"