Simplify error message formatting in test cases for data components

langflow-ai · Oct 22, 2024 · a438a1f · a438a1f
1 parent 0ac0d8a
commit a438a1f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/src/backend/tests/unit/components/prototypes/test_create_data_component.py b/src/backend/tests/unit/components/prototypes/test_create_data_component.py
@@ -1,4 +1,5 @@
 import pytest
+
 from langflow.components.prototypes.CreateData import CreateDataComponent
 from langflow.schema import Data
 
@@ -48,9 +49,7 @@ def test_update_build_config_exceed_limit(create_data_component):
             "value": False,
         },
     }
-    with pytest.raises(
-        ValueError, match="Number of fields cannot exceed 15. Try using a Component to combine two Data."
-    ):
+    with pytest.raises(ValueError, match="Number of fields cannot exceed 15."):
         create_data_component.update_build_config(build_config, 16, "number_of_fields")
 
 

diff --git a/src/backend/tests/unit/components/prototypes/test_update_data_component.py b/src/backend/tests/unit/components/prototypes/test_update_data_component.py
@@ -1,4 +1,5 @@
 import pytest
+
 from langflow.components.prototypes.UpdateData import UpdateDataComponent
 from langflow.schema import Data
 
@@ -48,9 +49,7 @@ def test_update_build_config_exceed_limit(update_data_component):
             "value": False,
         },
     }
-    with pytest.raises(
-        ValueError, match="Number of fields cannot exceed 15. Try using a Component to combine two Data."
-    ):
+    with pytest.raises(ValueError, match="Number of fields cannot exceed 15."):
         update_data_component.update_build_config(build_config, 16, "number_of_fields")
 
 
@@ -100,6 +99,6 @@ def test_validate_text_key_invalid(update_data_component):
         update_data_component.validate_text_key(data)
 
     expected_error_message = (
-        f"Text Key: {update_data_component.text_key} not found in the Data keys: {','.join(data.data.keys())}"
+        f"Text Key: '{update_data_component.text_key}' not found in the Data keys: {', '.join(data.data.keys())}"
     )
     assert str(exc_info.value) == expected_error_message