wehub-resource-sync
a7d6d88f6f
CI / changes (push) Has been cancelled
CI / cd libs/checkpoint (push) Has been cancelled
CI / cd libs/checkpoint-conformance (push) Has been cancelled
CI / cd libs/checkpoint-postgres (push) Has been cancelled
CI / cd libs/checkpoint-sqlite (push) Has been cancelled
CI / cd libs/cli (push) Has been cancelled
CI / cd libs/prebuilt (push) Has been cancelled
CI / cd libs/sdk-py (push) Has been cancelled
CI / cd libs/langgraph (push) Has been cancelled
CI / Check SDK methods matching (push) Has been cancelled
CI / Check CLI schema hasn't changed #3.13 (push) Has been cancelled
CI / CLI integration test (push) Has been cancelled
CI / sdk-py integration test (push) Has been cancelled
CI / CI Success (push) Has been cancelled
baseline / benchmark (push) Has been cancelled
Deploy Redirects to GitHub Pages / deploy (push) Has been cancelled
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
from langgraph._internal._constants import PULL, PUSH
|
|
from langgraph.pregel._algo import prepare_next_tasks, task_path_str
|
|
from langgraph.pregel._checkpoint import channels_from_checkpoint, empty_checkpoint
|
|
|
|
|
|
def test_prepare_next_tasks() -> None:
|
|
config = {}
|
|
processes = {}
|
|
checkpoint = empty_checkpoint()
|
|
channels, managed = channels_from_checkpoint({}, checkpoint)
|
|
|
|
assert (
|
|
prepare_next_tasks(
|
|
checkpoint,
|
|
{},
|
|
processes,
|
|
channels,
|
|
managed,
|
|
config,
|
|
0,
|
|
-1,
|
|
for_execution=False,
|
|
)
|
|
== {}
|
|
)
|
|
assert (
|
|
prepare_next_tasks(
|
|
checkpoint,
|
|
{},
|
|
processes,
|
|
channels,
|
|
managed,
|
|
config,
|
|
0,
|
|
-1,
|
|
for_execution=True,
|
|
checkpointer=None,
|
|
store=None,
|
|
manager=None,
|
|
)
|
|
== {}
|
|
)
|
|
|
|
# TODO: add more tests
|
|
|
|
|
|
def test_tuple_str() -> None:
|
|
push_path_a = (PUSH, 2)
|
|
pull_path_a = (PULL, "abc")
|
|
push_path_b = (PUSH, push_path_a, 1)
|
|
push_path_c = (PUSH, push_path_b, 3)
|
|
|
|
assert task_path_str(push_path_a) == f"~{PUSH}, 0000000002"
|
|
assert task_path_str(push_path_b) == f"~{PUSH}, ~{PUSH}, 0000000002, 0000000001"
|
|
assert (
|
|
task_path_str(push_path_c)
|
|
== f"~{PUSH}, ~{PUSH}, ~{PUSH}, 0000000002, 0000000001, 0000000003"
|
|
)
|
|
assert task_path_str(pull_path_a) == f"~{PULL}, abc"
|
|
|
|
path_list = [push_path_b, push_path_a, pull_path_a, push_path_c]
|
|
assert sorted(map(task_path_str, path_list)) == [
|
|
f"~{PULL}, abc",
|
|
f"~{PUSH}, 0000000002",
|
|
f"~{PUSH}, ~{PUSH}, 0000000002, 0000000001",
|
|
f"~{PUSH}, ~{PUSH}, ~{PUSH}, 0000000002, 0000000001, 0000000003",
|
|
]
|