test_inmemory_get_by_ids() — langchain Function Reference

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

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_inmemory_get_by_ids() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_in_memory.py lines 182–206

async def test_inmemory_get_by_ids() -> None:
    """Test get by ids."""
    store = InMemoryVectorStore(embedding=DeterministicFakeEmbedding(size=3))

    store.add_documents(
        [
            Document(page_content="foo", id="1", metadata={"metadata": "value"}),
            Document(page_content="bar", id="2"),
            Document(page_content="baz", id="3"),
        ],
    )

    # Check sync version
    output = store.get_by_ids(["1", "2"])
    assert output == [
        Document(page_content="foo", id="1", metadata={"metadata": "value"}),
        Document(page_content="bar", id="2"),
    ]

    # Check async version
    output = await store.aget_by_ids(["1", "3", "5"])
    assert output == [
        Document(page_content="foo", id="1", metadata={"metadata": "value"}),
        Document(page_content="baz", id="3"),
    ]