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

Add PythonFunctionComponent and test case for it

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 00:13:33 -03:00
commit a8779b1205
6 changed files with 56 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_helper_components.py
View file

@ -1,5 +1,4 @@
from langchain_core.documents import Document
from langflow.components import helpers
from langflow.interface.custom.utils import build_custom_component_template
from langflow.schema import Record
@ -52,23 +51,9 @@ def test_uuid_generator_component():
assert len(result) == 36
def test_python_function_component():
# Arrange
python_function_component = helpers.PythonFunctionComponent()
# Act
# function must be a string representation
function = "def function():\n return 'Hello, World!'"
# result is the callable function
result = python_function_component.build(function)
# Assert
assert result() == "Hello, World!"
def test_records_as_text_component():
# Arrange
records_as_text_component = helpers.RecordsAsTextComponent()
records_as_text_component = helpers.RecordsToTextComponent()
# Act
# Replace with your actual test data
@ -83,7 +68,7 @@ def test_records_as_text_component():
def test_text_to_record_component():
# Arrange
text_to_record_component = helpers.TextToRecordComponent()
text_to_record_component = helpers.CreateRecordComponent()
# Act
# Replace with your actual test data