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

Refactor test_custom_component.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-22 10:39:59 -03:00
commit 113861eb59
1 changed files with 6 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/test_custom_component.py
View file

@ -5,12 +5,9 @@ from uuid import uuid4
import pytest import pytest
from fastapi import HTTPException from fastapi import HTTPException
from langflow.interface.custom.base import CustomComponent from langflow.interface.custom.base import CustomComponent
from langflow.interface.custom.code_parser.code_parser import (CodeParser, from langflow.interface.custom.code_parser.code_parser import CodeParser, CodeSyntaxError
CodeSyntaxError) from langflow.interface.custom.custom_component.component import Component, ComponentCodeNullError
from langflow.interface.custom.custom_component.component import ( from langflow.interface.custom.utils import build_custom_component_template
Component, ComponentCodeNullError)
from langflow.interface.custom.utils import (build_custom_component_template,
create_and_validate_component)
from langflow.services.database.models.flow import Flow, FlowCreate from langflow.services.database.models.flow import Flow, FlowCreate
code_default = """ code_default = """
@ -535,12 +532,12 @@ def test_build_config_field_value_keys(component):
def test_create_and_validate_component_valid_code(test_component_code): def test_create_and_validate_component_valid_code(test_component_code):
component = create_and_validate_component(test_component_code) component = CustomComponent(code=test_component_code)
assert isinstance(component, CustomComponent) assert isinstance(component, CustomComponent)
def test_build_langchain_template_custom_component_valid_code(test_component_code): def test_build_langchain_template_custom_component_valid_code(test_component_code):
component = create_and_validate_component(test_component_code) component = CustomComponent(code=test_component_code)
frontend_node = build_custom_component_template(component) frontend_node = build_custom_component_template(component)
assert isinstance(frontend_node, dict) assert isinstance(frontend_node, dict)
template = frontend_node["template"] template = frontend_node["template"]
@ -554,7 +551,7 @@ def test_build_langchain_template_custom_component_valid_code(test_component_cod
def test_build_langchain_template_custom_component_templatefield(test_component_with_templatefield_code): def test_build_langchain_template_custom_component_templatefield(test_component_with_templatefield_code):
component = create_and_validate_component(test_component_with_templatefield_code) component = CustomComponent(code=test_component_with_templatefield_code)
frontend_node = build_custom_component_template(component) frontend_node = build_custom_component_template(component)
assert isinstance(frontend_node, dict) assert isinstance(frontend_node, dict)
template = frontend_node["template"] template = frontend_node["template"]