WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-17 21:25:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

more explicit tests

Browse Source
This commit is contained in:
Sydney Runkle
2025-04-23 11:49:26 -07:00
parent d6b4ee348f
commit 8fe7c8101f
2 changed files with 65 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_pregel.py
+37 -3
View File
@@ -5422,7 +5422,15 @@ def test_interrupt_functional(
config = {"configurable": {"thread_id": "1"}}
# First run, interrupted at bar
graph.invoke({"a": ""}, config)
assert graph.invoke({"a": ""}, config) == {
"__interrupt__": [
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:")],
)
]
}
# Resume with an answer
res = graph.invoke(Command(resume="bar"), config)
assert res == {"a": "foobar", "b": "bar"}
@@ -5453,7 +5461,20 @@ def test_interrupt_task_functional(
config = {"configurable": {"thread_id": "1"}}
# First run, interrupted at bar
graph.invoke({"a": ""}, config)
assert graph.invoke({"a": ""}, config) == {
"__interrupt__": [
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
]
}
# Resume with an answer
res = graph.invoke(Command(resume="bar"), config)
assert res == {"a": "foobar"}
@@ -5469,7 +5490,20 @@ def test_interrupt_task_functional(
return baz_result
# First run, interrupted at bar
graph.invoke({"a": ""}, config)
assert graph.invoke({"a": ""}, config) == {
"__interrupt__": [
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
]
}
# Provide resumes
graph.invoke(Command(resume="bar"), config)
assert graph.invoke(Command(resume="baz"), config) == {"a": "foobarbaz"}
libs/langgraph/tests/test_pregel_async.py
+28 -2
View File
@@ -8226,7 +8226,20 @@ async def test_handles_multiple_interrupts_from_tasks() -> None:
config = {"configurable": {"thread_id": "1"}}
result = await program.ainvoke("this is ignored", config=config)
assert "__interrupt__" in result
assert result == {
"__interrupt__": [
Interrupt(
value="Hey do you want to add James?",
resumable=True,
ns=[AnyStr("program:"), AnyStr("add_participant:")],
),
Interrupt(
value="Hey do you want to add James?",
resumable=True,
ns=[AnyStr("program:"), AnyStr("add_participant:")],
),
]
}
state = await program.aget_state(config=config)
assert len(state.tasks[0].interrupts) == 1
@@ -8238,7 +8251,20 @@ async def test_handles_multiple_interrupts_from_tasks() -> None:
assert task_interrupt.value == "Hey do you want to add James?"
result = await program.ainvoke(Command(resume=True), config=config)
assert "__interrupt__" in result
assert result == {
"__interrupt__": [
Interrupt(
value="Hey do you want to add Will?",
resumable=True,
ns=[AnyStr("program:"), AnyStr("add_participant:")],
),
Interrupt(
value="Hey do you want to add Will?",
resumable=True,
ns=[AnyStr("program:"), AnyStr("add_participant:")],
),
]
}
state = await program.aget_state(config=config)
assert len(state.tasks[0].interrupts) == 1