jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

test: Fix assertion in test_get_vertices function. (#3387)

Browse source 
* test: Fix assertion in test_get_vertices function.

* test: fix set

---------

Co-authored-by: italojohnny <italojohnnydosanjos@gmail.com>
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-16 14:49:02 -03:00 • committed by GitHub
commit e4e1a747f8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/tests/test_endpoints.py
View file

@ -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):