diff --git a/tests/test_pregel.py b/tests/test_pregel.py index 392ce7c80..ad5c32a09 100644 --- a/tests/test_pregel.py +++ b/tests/test_pregel.py @@ -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 diff --git a/tests/test_pregel_async.py b/tests/test_pregel_async.py index 2c84b34c2..df1c1c024 100644 --- a/tests/test_pregel_async.py +++ b/tests/test_pregel_async.py @@ -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