From ef4b9f96e55f8063e81d923dd394b8ec5cb802ec Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Mar 2024 14:27:23 -0300 Subject: [PATCH] Refactor code for improved readability and maintainability --- tests/test_helper_components.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_helper_components.py b/tests/test_helper_components.py index 28b84a42e..82515b699 100644 --- a/tests/test_helper_components.py +++ b/tests/test_helper_components.py @@ -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)