test_runnable_branch_ainvoke() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_runnable_branch_ainvoke() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 4574–4599

async def test_runnable_branch_ainvoke() -> None:
    """Test async variant of invoke."""
    branch = RunnableBranch[int, int](
        (lambda x: x > 0 and x < 5, lambda x: x + 1),
        (lambda x: x > 5, lambda x: x * 10),
        lambda x: x - 1,
    )

    assert await branch.ainvoke(1) == 2
    assert await branch.ainvoke(10) == 100
    assert await branch.ainvoke(0) == -1

    # Verify that the async variant is used if available
    async def condition(x: int) -> bool:
        return x > 0

    async def add(x: int) -> int:
        return x + 1

    async def sub(x: int) -> int:
        return x - 1

    branch = RunnableBranch[int, int]((condition, add), sub)

    assert await branch.ainvoke(1) == 2
    assert await branch.ainvoke(-10) == -11