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

📝 (tests/conftest.py): Remove duplicate imports and organize imports for better readability

Browse source 
♻️ (tests/test_custom_component.py): Refactor CustomComponent to Component for better naming consistency
♻️ (tests/test_endpoints.py): Refactor test functions to improve readability and maintainability by simplifying assertions and organizing code
This commit is contained in:
ogabrielluiz 2024-06-04 09:25:27 -03:00
commit e4f4401d75
3 changed files with 23 additions and 16 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 langchain_core.documents import Document
from langflow.custom import CustomComponent
from langflow.custom import CustomComponent, Component
from langflow.custom.code_parser.code_parser import CodeParser, CodeSyntaxError
from langflow.custom.custom_component.base_component import BaseComponent, ComponentCodeNullError
from langflow.custom.utils import build_custom_component_template
@ -15,7 +15,7 @@ from langflow.services.database.models.flow import Flow, FlowCreate
def code_component_with_multiple_outputs():
with open("tests/data/component_multiple_outputs.py", "r") as f:
code = f.read()
return CustomComponent(code=code)
return Component(code=code)
code_default = """