From 4b8442896b9adefa6536255df6c995ab57fd27de Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 29 Sep 2023 16:50:00 +0100 Subject: [PATCH] Make test deterministic --- .../tests/unit_tests/schema/runnable/test_runnable.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py index 4a63f92ff2..9d1d131f0d 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py @@ -582,7 +582,9 @@ async def test_with_config(mocker: MockerFixture) -> None: ) == [5, 7] assert len(spy.call_args_list) == 2 - for i, call in enumerate(spy.call_args_list): + for i, call in enumerate( + sorted(spy.call_args_list, key=lambda x: 0 if x.args[0] == "hello" else 1) + ): assert call.args[0] == ("hello" if i == 0 else "wooorld") if i == 0: assert call.args[1].get("recursion_limit") == 5