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

Fix return type assertion in

Browse source 
test_custom_component_get_function_entrypoint_return_type()
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 12:56:10 -03:00
commit 5c352157d1
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_custom_component.py
View file

@ -4,7 +4,7 @@ from uuid import uuid4
import pytest
from fastapi import HTTPException
from langflow.field_typing.constants import Data
from langflow.interface.custom.base import CustomComponent
from langflow.interface.custom.code_parser import CodeParser, CodeSyntaxError
from langflow.interface.custom.component import Component, ComponentCodeNullError
@ -412,7 +412,7 @@ class MyClass(CustomComponent):
custom_component = CustomComponent(code=my_code, function_entrypoint_name="build")
return_type = custom_component.get_function_entrypoint_return_type
assert return_type == [Data]
assert return_type == []
def test_custom_component_get_main_class_name_no_main_class():