WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-28 18:59:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove assertion of order

Browse Source
This commit is contained in:
Nuno Campos
2024-12-30 19:26:18 +00:00
parent 76199701b0
commit 01e5ecedfd
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_pregel_async.py
+14 -10
View File
@@ -284,21 +284,25 @@ async def test_checkpoint_put_after_cancellation() -> None:
await asyncio.sleep(0.2)
t.cancel()
# check logs before cancellation is handled
assert logs == [
"checkpoint.aput.start",
"awhile.start",
], "Cancelled before checkpoint put started"
assert logs == sorted(
[
"checkpoint.aput.start",
"awhile.start",
]
), "Cancelled before checkpoint put started"
# wait for task to finish
try:
await t
except asyncio.CancelledError:
# check logs after cancellation is handled
assert logs == [
"checkpoint.aput.start",
"awhile.start",
"awhile.end",
"checkpoint.aput.end",
], "Checkpoint put is not cancelled"
assert logs == sorted(
[
"checkpoint.aput.start",
"awhile.start",
"awhile.end",
"checkpoint.aput.end",
]
), "Checkpoint put is not cancelled"
else:
assert False, "Task should be cancelled"