21 lines
486 B
Python
21 lines
486 B
Python
from langflow.components import prototypes
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
pass
|
|
|
|
|
|
def test_python_function_component():
|
|
# Arrange
|
|
python_function_component = prototypes.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!"
|