Spaces:
Running
Running
File size: 872 Bytes
d202ada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import pytest
from langflow.custom.custom_component.custom_component import CustomComponent
from langflow.field_typing.constants import Data
@pytest.fixture
def component(
client, # noqa: ARG001
active_user,
):
return CustomComponent(
user_id=active_user.id,
field_config={
"fields": {
"llm": {"type": "str"},
"url": {"type": "str"},
"year": {"type": "int"},
}
},
)
def test_list_flows_flow_objects(component):
flows = component.list_flows()
are_flows = [isinstance(flow, Data) for flow in flows]
flow_types = [type(flow) for flow in flows]
assert all(are_flows), f"Expected all flows to be Data objects, got {flow_types}"
def test_list_flows_return_type(component):
flows = component.list_flows()
assert isinstance(flows, list)
|