From 16c86c9de6acb6806bb93262bf907f0f18b1db11 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 21 Jan 2025 09:49:46 -0800 Subject: [PATCH] Add test --- libs/langgraph/tests/test_pregel_async.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py index 17847f4b8..e937b1949 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -2475,7 +2475,12 @@ async def test_imp_task(checkpointer_name: str) -> None: assert mapper_calls == 2 assert len(tracer.runs) == 1 assert len(tracer.runs[0].child_runs) == 1 - assert tracer.runs[0].child_runs[0].name == "graph" + entrypoint_run = tracer.runs[0].child_runs[0] + assert entrypoint_run.name == "graph" + mapper_runs = [r for r in entrypoint_run.child_runs if r.name == "mapper"] + assert len(mapper_runs) == 2 + assert mapper_runs[0].inputs == {"input": 0} + assert mapper_runs[1].inputs == {"input": 1} assert await graph.ainvoke(Command(resume="answer"), thread1) == [ "00answer",