test_ayield_keys() — langchain Function Reference

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

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_ayield_keys() — langchain Function Reference

Dependency Diagram

graph TD
  d6b4927c_8862_d381_875a_79300f6d8392["test_ayield_keys()"]
  78341c80_3d5a_da08_2c6f_25554ec126bf["test_in_memory.py"]
  d6b4927c_8862_d381_875a_79300f6d8392 -->|defined in| 78341c80_3d5a_da08_2c6f_25554ec126bf
  style d6b4927c_8862_d381_875a_79300f6d8392 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/stores/test_in_memory.py lines 118–129

async def test_ayield_keys() -> None:
    store = InMemoryStore()
    await store.amset([("key1", "value1"), ("key2", "value2"), ("key3", "value3")])

    keys = [key async for key in store.ayield_keys()]
    assert set(keys) == {"key1", "key2", "key3"}

    keys_with_prefix = [key async for key in store.ayield_keys(prefix="key")]
    assert set(keys_with_prefix) == {"key1", "key2", "key3"}

    keys_with_invalid_prefix = [key async for key in store.ayield_keys(prefix="x")]
    assert keys_with_invalid_prefix == []