jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update DocumentsToDataComponent class name in test_helper_components.py

Browse source 
The code changes in test_helper_components.py update the class name from DocumentToDataComponent to DocumentsToDataComponent. This refactor improves the clarity and consistency of the code.
This commit is contained in:
ogabrielluiz 2024-06-18 22:42:06 -03:00
commit 412a793a4c
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/unit/test_helper_components.py
View file

@ -1,5 +1,4 @@
from langchain_core.documents import Document from langchain_core.documents import Document
from langflow.components import helpers from langflow.components import helpers
from langflow.custom.utils import build_custom_component_template from langflow.custom.utils import build_custom_component_template
from langflow.schema import Data from langflow.schema import Data
@ -20,7 +19,7 @@ def test_update_data_component():
def test_document_to_data_component(): def test_document_to_data_component():
# Arrange # Arrange
document_to_data_component = helpers.DocumentToDataComponent() document_to_data_component = helpers.DocumentsToDataComponent()
# Act # Act
# Replace with your actual test data # Replace with your actual test data
@ -54,17 +53,18 @@ def test_uuid_generator_component():
def test_data_as_text_component(): def test_data_as_text_component():
# Arrange # Arrange
data_as_text_component = helpers.DataToTextComponent() data_as_text_component = helpers.ParseDataComponent()
# Act # Act
# Replace with your actual test data # Replace with your actual test data
data = [Data(data={"key": "value", "bacon": "eggs"})] data = [Data(data={"key": "value", "bacon": "eggs"})]
template = "Data:{data} -- Bacon:{bacon}" template = "Data:{data} -- Bacon:{bacon}"
result = data_as_text_component.build(data, template=template) data_as_text_component.set_attributes({"data": data, "template": template})
result = data_as_text_component.parse_data()
# Assert # Assert
# Replace with your actual expected result # Replace with your actual expected result
assert result == "Data:{'key': 'value', 'bacon': 'eggs'} -- Bacon:eggs" assert result.text == "Data:{'key': 'value', 'bacon': 'eggs'} -- Bacon:eggs"
def test_text_to_data_component(): def test_text_to_data_component():