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

Add test for build ordering in complex graph

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-20 16:16:30 -03:00
commit 7ae3aefa76
2 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_graph.py
View file

@ -430,3 +430,15 @@ async def test_pickle_each_vertex(json_vector_store):
assert pickled is not None
unpickled = pickle.loads(pickled)
assert unpickled is not None
@pytest.mark.asyncio
async def test_build_ordering(complex_graph_with_groups):
sorted_vertices = complex_graph_with_groups.sort_vertices(stop_component_id="ChatInput-Ay8QQ")
assert sorted_vertices == [
"ChatInput-Ay8QQ",
"RecordsAsText-vkx2A",
"FileLoader-Vo1Cq",
]
sorted_vertices = complex_graph_with_groups.sort_vertices()