Refactor test_helper_components.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-09 23:39:43 -03:00
commit d4313a05ce

View file

@ -24,9 +24,7 @@ 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
@ -44,9 +42,7 @@ def test_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)