test_runnable_assign() — langchain Function Reference

Architecture documentation for the test_runnable_assign() function in test_runnable.py from the langchain codebase.

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_runnable_assign() — langchain Function Reference

Dependency Diagram

graph TD
  aadb633a_81ad_97c4_e0f0_a000f82bfd28["test_runnable_assign()"]
  26df6ad8_0189_51d0_c3c1_6c3248893ff5["test_runnable.py"]
  aadb633a_81ad_97c4_e0f0_a000f82bfd28 -->|defined in| 26df6ad8_0189_51d0_c3c1_6c3248893ff5
  fb618d44_c03b_ea8b_385b_2278dfb173d4["invoke()"]
  aadb633a_81ad_97c4_e0f0_a000f82bfd28 -->|calls| fb618d44_c03b_ea8b_385b_2278dfb173d4
  style aadb633a_81ad_97c4_e0f0_a000f82bfd28 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 5725–5733

def test_runnable_assign() -> None:
    def add_ten(x: dict[str, int]) -> dict[str, int]:
        return {"added": x["input"] + 10}

    mapper = RunnableParallel({"add_step": RunnableLambda(add_ten)})
    runnable_assign = RunnableAssign(mapper)

    result = runnable_assign.invoke({"input": 5})
    assert result == {"input": 5, "add_step": {"added": 15}}