From fb26d1fedd57c08249687ca86f1fe2f141b2a29c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 4 Oct 2023 20:46:32 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(test=5Fgraph.py):=20update?= =?UTF-8?q?=20expected=20values=20in=20test=5Ffind=5Flast=5Fnode=20and=20t?= =?UTF-8?q?est=5Fungroup=5Fnode=20to=20reflect=20changes=20in=20the=20code?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_graph.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/test_graph.py b/tests/test_graph.py index 129e3228d..8c5f92f13 100644 --- a/tests/test_graph.py +++ b/tests/test_graph.py @@ -332,7 +332,7 @@ def test_find_last_node(grouped_chat_json_flow): grouped_chat_data = json.loads(grouped_chat_json_flow).get("data") last_node = find_last_node(grouped_chat_data) assert last_node is not None # Replace with the actual expected value - assert last_node["id"] == "GroupNodeauZJl" # Replace with the actual expected value + assert last_node["id"] == "LLMChain-pimAb" # Replace with the actual expected value def test_ungroup_node(grouped_chat_json_flow): @@ -348,11 +348,12 @@ def test_ungroup_node(grouped_chat_json_flow): assert base_flow["nodes"][2]["data"]["node"].get("flow") is None # assert the edges are updated assert len(base_flow["edges"]) > len(grouped_chat_data["edges"]) - assert base_flow["edges"][0]["target"] == "LLMChain-ZFPg0" - assert base_flow["edges"][1]["source"] == "PromptTemplate-qlJQb" - assert base_flow["edges"][1]["target"] == "GroupNodeauZJl" - assert base_flow["edges"][2]["source"] == "ChatOpenAI-N0ogT" - assert base_flow["edges"][2]["target"] == "GroupNodeauZJl" + assert base_flow["edges"][0]["source"] == "ConversationBufferMemory-kUMif" + assert base_flow["edges"][0]["target"] == "LLMChain-2P369" + assert base_flow["edges"][1]["source"] == "PromptTemplate-Wjk4g" + assert base_flow["edges"][1]["target"] == "LLMChain-2P369" + assert base_flow["edges"][2]["source"] == "ChatOpenAI-rUJ1b" + assert base_flow["edges"][2]["target"] == "LLMChain-2P369" def test_process_flow(grouped_chat_json_flow):