Update tests

This commit is contained in:
Nuno Campos
2025-01-15 13:35:51 -08:00
parent 3626478029
commit c2a57385c0
2 changed files with 72 additions and 12 deletions
+36 -6
View File
@@ -197,7 +197,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": False,
"__pregel_store": None,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": None,
"checkpoint_map": {
@@ -264,7 +269,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": False,
"__pregel_store": None,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -361,7 +371,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": False,
"__pregel_store": None,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -468,7 +483,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": True,
"__pregel_store": None,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": None,
"checkpoint_map": {
@@ -530,7 +550,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": True,
"__pregel_store": None,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -648,7 +673,12 @@ async def test_subgraph_w_interrupt(
"__pregel_resuming": True,
"__pregel_store": None,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -196,7 +196,12 @@ def test_subgraph_w_interrupt(
"__pregel_resuming": False,
"__pregel_store": None,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": None,
"checkpoint_map": {
@@ -263,7 +268,12 @@ def test_subgraph_w_interrupt(
"__pregel_dedupe_tasks": True,
"__pregel_resuming": False,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -360,7 +370,12 @@ def test_subgraph_w_interrupt(
"__pregel_store": None,
"__pregel_resuming": False,
"__pregel_task_id": history[0].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -466,7 +481,12 @@ def test_subgraph_w_interrupt(
"__pregel_store": None,
"__pregel_resuming": True,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": None,
"checkpoint_map": {
@@ -528,7 +548,12 @@ def test_subgraph_w_interrupt(
"__pregel_store": None,
"__pregel_resuming": True,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {
@@ -646,7 +671,12 @@ def test_subgraph_w_interrupt(
"__pregel_resuming": True,
"__pregel_store": None,
"__pregel_task_id": history[1].tasks[0].id,
"__pregel_scratchpad": {},
"__pregel_scratchpad": {
"call_counter": 0,
"interrupt_counter": -1,
"null_resume": None,
"resume": [],
},
"__pregel_writes": AnyList(),
"checkpoint_id": c.config["configurable"]["checkpoint_id"],
"checkpoint_map": {