From e4e1a747f820af351b4517ea0c84f8264be6672f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 16 Aug 2024 14:49:02 -0300 Subject: [PATCH] test: Fix assertion in test_get_vertices function. (#3387) * test: Fix assertion in test_get_vertices function. * test: fix set --------- Co-authored-by: italojohnny --- src/backend/tests/test_endpoints.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/tests/test_endpoints.py b/src/backend/tests/test_endpoints.py index 2c4a32c8c..a57e06802 100644 --- a/src/backend/tests/test_endpoints.py +++ b/src/backend/tests/test_endpoints.py @@ -407,11 +407,12 @@ def test_get_vertices(client, added_flow_with_prompt_and_history, logged_in_head # ['ConversationBufferMemory-Lu2Nb', 'PromptTemplate-5Q0W8', 'ChatOpenAI-vy7fV', 'LLMChain-UjBh1'] # The important part is before the - (ConversationBufferMemory, PromptTemplate, ChatOpenAI, LLMChain) ids = [_id.split("-")[0] for _id in response.json()["ids"]] - assert ids == [ + + assert set(ids) == { "ChatOpenAI", "PromptTemplate", "ConversationBufferMemory", - ] + } def test_build_vertex_invalid_flow_id(client, logged_in_headers):