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

refac: implemented buid_json using graph methods

Browse source
This commit is contained in:
Gabriel Almeida 2023-03-24 12:22:27 -03:00
commit 230f0d95e9
3 changed files with 28 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_graph.py
View file

@ -17,14 +17,14 @@ def get_graph(basic=True):
return Graph(nodes, edges)
def test_get_connected_nodes():
def test_get_nodes_with_target():
"""Test getting connected nodes"""
graph = get_graph()
assert isinstance(graph, Graph)
# Get root node
root = get_root_node(graph)
assert root is not None
connected_nodes = graph.get_connected_nodes(root)
connected_nodes = graph.get_nodes_with_target(root)
assert connected_nodes is not None