pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 07:34:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor code for improved readability and maintainability

Browse Source
This commit is contained in:
Gabriel Luiz Freitas Almeida
2024-03-11 14:27:23 -03:00
parent 9802322ae9
commit ef4b9f96e5
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_helper_components.py
View File

@ -24,7 +24,9 @@ def test_document_to_record_component():
# Act
# Replace with your actual test data
document = Document(page_content="key: value", metadata={"url": "https://example.com"})
document = Document(
page_content="key: value", metadata={"url": "https://example.com"}
)
result = document_to_record_component.build(document)
# Assert
@ -37,12 +39,14 @@ def test_uuid_generator_component():
uuid_generator_component = helpers.UUIDGeneratorComponent()
uuid_generator_component.code = open(helpers.IDGenerator.__file__, "r").read()
frontend_node = build_custom_component_template(uuid_generator_component)
frontend_node, _ = build_custom_component_template(uuid_generator_component)
# Act
build_config = frontend_node.get("template")
field_name = "unique_id"
build_config = uuid_generator_component.update_build_config(build_config, None, field_name)
build_config = uuid_generator_component.update_build_config(
build_config, None, field_name
)
unique_id = build_config["unique_id"]["value"]
result = uuid_generator_component.build(unique_id)