test_yield_keys() — langchain Function Reference

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

Function python LangChainCore Runnables calls 1 called by 1

Entity Profile

LangChainCore→ Runnables→ test_yield_keys() — langchain Function Reference

Dependency Diagram

graph TD
  727944bb_67cf_0009_4883_56e1a0b77347["test_yield_keys()"]
  6a8c29e9_dfc7_44bd_8ca2_6c43a1b6eb98["BaseStoreAsyncTests"]
  727944bb_67cf_0009_4883_56e1a0b77347 -->|defined in| 6a8c29e9_dfc7_44bd_8ca2_6c43a1b6eb98
  1185aafb_a3db_e8ab_6436_11a631876502["test_yield_keys()"]
  1185aafb_a3db_e8ab_6436_11a631876502 -->|calls| 727944bb_67cf_0009_4883_56e1a0b77347
  1185aafb_a3db_e8ab_6436_11a631876502["test_yield_keys()"]
  727944bb_67cf_0009_4883_56e1a0b77347 -->|calls| 1185aafb_a3db_e8ab_6436_11a631876502
  style 727944bb_67cf_0009_4883_56e1a0b77347 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/standard-tests/langchain_tests/integration_tests/base_store.py lines 299–315

    async def test_yield_keys(
        self,
        kv_store: BaseStore[str, V],
        three_values: tuple[V, V, V],
    ) -> None:
        """Test that we can yield keys from the store."""
        foo, bar, _buzz = three_values
        key_value_pairs = [("foo", foo), ("bar", bar)]
        await kv_store.amset(key_value_pairs)

        generator = kv_store.ayield_keys()
        assert isinstance(generator, AsyncGenerator)

        assert sorted([key async for key in kv_store.ayield_keys()]) == ["bar", "foo"]
        assert sorted([key async for key in kv_store.ayield_keys(prefix="foo")]) == [
            "foo",
        ]