WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-22 01:25:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

rename node name in tests

Browse Source
This commit is contained in:
hmasdev
2024-04-11 20:34:45 +09:00
parent 42de3304dc
commit 236ecb1982
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_pregel.py
+3 -3
View File
@@ -74,9 +74,9 @@ def test_invoke_single_process_in_out(mocker: MockerFixture) -> None:
)
def test_invoke_single_process_in_out_falsy_values(falsy_value: Any) -> None:
graph = Graph()
graph.add_node("add_one", lambda *args, **kwargs: falsy_value)
graph.set_entry_point("add_one")
graph.set_finish_point("add_one")
graph.add_node("return_falsy_const", lambda *args, **kwargs: falsy_value)
graph.set_entry_point("return_falsy_const")
graph.set_finish_point("return_falsy_const")
gapp = graph.compile()
assert gapp.invoke(1, allow_falsy_output=True) == falsy_value
assert gapp.invoke(1) is None
tests/test_pregel_async.py
+3 -3
View File
@@ -71,9 +71,9 @@ async def test_invoke_single_process_in_out(mocker: MockerFixture) -> None:
)
async def test_invoke_single_process_in_out_falsy_values(falsy_value: Any) -> None:
graph = Graph()
graph.add_node("add_one", lambda *args, **kwargs: falsy_value)
graph.set_entry_point("add_one")
graph.set_finish_point("add_one")
graph.add_node("return_falsy_const", lambda *args, **kwargs: falsy_value)
graph.set_entry_point("return_falsy_const")
graph.set_finish_point("return_falsy_const")
gapp = graph.compile()
assert (await gapp.ainvoke(1, allow_falsy_output=True)) == falsy_value
assert (await gapp.ainvoke(1)) is None