test_yield_keys() — langchain Function Reference

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

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_yield_keys() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/stores/test_in_memory.py lines 104–115

def test_yield_keys() -> None:
    store = InMemoryStore()
    store.mset([("key1", "value1"), ("key2", "value2"), ("key3", "value3")])

    keys = list(store.yield_keys())
    assert set(keys) == {"key1", "key2", "key3"}

    keys_with_prefix = list(store.yield_keys(prefix="key"))
    assert set(keys_with_prefix) == {"key1", "key2", "key3"}

    keys_with_invalid_prefix = list(store.yield_keys(prefix="x"))
    assert keys_with_invalid_prefix == []