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

Refactor test_loading.py to use Graph instead of Chain

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 10:23:55 -03:00
commit c7ad808c9e
1 changed files with 3 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/test_loading.py
View file

@ -1,17 +1,14 @@
import json
import pytest import pytest
from langchain.chains.base import Chain
from langflow.graph import Graph from langflow.graph import Graph
from langflow.processing.load import load_flow_from_json from langflow.processing.load import load_flow_from_json
from langflow.utils.payload import get_root_vertex
def test_load_flow_from_json(): def test_load_flow_from_json():
"""Test loading a flow from a json file""" """Test loading a flow from a json file"""
loaded = load_flow_from_json(pytest.BASIC_EXAMPLE_PATH) loaded = load_flow_from_json(pytest.BASIC_EXAMPLE_PATH)
assert loaded is not None assert loaded is not None
assert isinstance(loaded, Chain) assert isinstance(loaded, Graph)
def test_load_flow_from_json_with_tweaks(): def test_load_flow_from_json_with_tweaks():
@ -19,18 +16,5 @@ def test_load_flow_from_json_with_tweaks():
tweaks = {"dndnode_82": {"model_name": "gpt-3.5-turbo-16k-0613"}} tweaks = {"dndnode_82": {"model_name": "gpt-3.5-turbo-16k-0613"}}
loaded = load_flow_from_json(pytest.BASIC_EXAMPLE_PATH, tweaks=tweaks) loaded = load_flow_from_json(pytest.BASIC_EXAMPLE_PATH, tweaks=tweaks)
assert loaded is not None assert loaded is not None
assert isinstance(loaded, Chain) assert isinstance(loaded, Graph)
assert loaded.llm.model_name == "gpt-3.5-turbo-16k-0613" assert loaded.llm.model_name == "gpt-3.5-turbo-16k-0613"
def test_get_root_vertex():
with open(pytest.BASIC_EXAMPLE_PATH, "r") as f:
flow_graph = json.load(f)
data_graph = flow_graph["data"]
nodes = data_graph["nodes"]
edges = data_graph["edges"]
graph = Graph(nodes, edges)
root = get_root_vertex(graph)
assert root is not None
assert hasattr(root, "id")
assert hasattr(root, "data")