From 44099d27c80a5ebc2dbf27963f53defe8f4406c0 Mon Sep 17 00:00:00 2001
From: Sydney Runkle <sydneymarierunkle@gmail.com>
Date: Fri, 25 Apr 2025 16:53:37 -0700
Subject: [PATCH] fixing tests

---
 libs/langgraph/langgraph/pregel/algo.py           | 12 +++++++++++-
 libs/langgraph/tests/test_checkpoint_migration.py |  4 +++-
 libs/langgraph/tests/test_pregel.py               |  2 +-
 3 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/libs/langgraph/langgraph/pregel/algo.py b/libs/langgraph/langgraph/pregel/algo.py
index b855ccd7e..029cf73be 100644
--- a/libs/langgraph/langgraph/pregel/algo.py
+++ b/libs/langgraph/langgraph/pregel/algo.py
@@ -908,7 +908,17 @@ def _scratchpad(
             if not isinstance(task_resume_write, list):
                 task_resume_write = [task_resume_write]
         else:
-            task_resume_write = []
+            # find task-specific resume value
+            for w in pending_writes:
+                if w[0] == task_id and w[1] == RESUME:
+                    task_resume_write = w[2]
+                    if not isinstance(task_resume_write, list):
+                        task_resume_write = [task_resume_write]
+                    break
+            else:
+                task_resume_write = []
+            # clear var
+            del w
     else:
         null_resume_write = None
         task_resume_write = []
diff --git a/libs/langgraph/tests/test_checkpoint_migration.py b/libs/langgraph/tests/test_checkpoint_migration.py
index de39f279e..1ac86adba 100644
--- a/libs/langgraph/tests/test_checkpoint_migration.py
+++ b/libs/langgraph/tests/test_checkpoint_migration.py
@@ -114,7 +114,9 @@ def get_expected_history(*, exc_task_results: int = 0) -> list[StateSnapshot]:
                     else {"answer": "doc1,doc2,doc3,doc4"},
                 ),
             ),
-            interrupts=(
+            interrupts=()
+            if exc_task_results
+            else (
                 Interrupt(
                     value="",
                     resumable=True,
diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 3b60a8627..7d12b6b20 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/libs/langgraph/tests/test_pregel.py
@@ -6218,7 +6218,7 @@ def test_multiple_interrupts_functional(
         values = []
 
         for idx in [1, 2, 3]:
-            values.extend([double(idx).result(), interrupt({"a": "boo"})])
+            values.extend([double(idx).result(), interrupt({"a": f"boo + {idx}"})])
 
         return {"values": values}