From 5929b0b08d2d7c73b82f7a642927af7a550414fd Mon Sep 17 00:00:00 2001
From: Sydney Runkle <sydneymarierunkle@gmail.com>
Date: Wed, 23 Apr 2025 16:19:46 -0700
Subject: [PATCH] more test fixes

---
 libs/langgraph/tests/test_pregel.py       | 10 ----------
 libs/langgraph/tests/test_pregel_async.py | 10 +++++++++-
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 901041795..2e0b91ef3 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/libs/langgraph/tests/test_pregel.py
@@ -5468,11 +5468,6 @@ def test_interrupt_task_functional(
                 resumable=True,
                 ns=[AnyStr("graph:"), AnyStr("bar:")],
             ),
-            Interrupt(
-                value="Provide value for bar:",
-                resumable=True,
-                ns=[AnyStr("graph:"), AnyStr("bar:")],
-            ),
         ]
     }
     # Resume with an answer
@@ -5497,11 +5492,6 @@ def test_interrupt_task_functional(
                 resumable=True,
                 ns=[AnyStr("graph:"), AnyStr("bar:")],
             ),
-            Interrupt(
-                value="Provide value for bar:",
-                resumable=True,
-                ns=[AnyStr("graph:"), AnyStr("bar:")],
-            ),
         ]
     }
     # Provide resumes
diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index 2cb578b30..df470195c 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -6873,7 +6873,15 @@ async def test_interrupt_task_functional(checkpointer_name: str) -> None:
 
         config = {"configurable": {"thread_id": "1"}}
         # First run, interrupted at bar
-        await graph.ainvoke({"a": ""}, config)
+        assert graph.ainvoke({"a": ""}, config) == {
+            "__interrupt__": [
+                Interrupt(
+                    value="Provide value for bar:",
+                    resumable=True,
+                    ns=[AnyStr("graph:"), AnyStr("bar:")],
+                ),
+            ]
+        }
         # Resume with an answer
         res = await graph.ainvoke(Command(resume="bar"), config)
         assert res == {"a": "foobar"}