fix: address review feedback for processing and serialization test files

Author: ogabrielluiz

src/backend/tests/unit/processing/test_expand_flow.py

@@ -1,7 +1,6 @@
 """Tests for langflow.processing.expand_flow module."""
 
 import pytest
-
 from langflow.processing.expand_flow import (
     CompactEdge,
     CompactFlowData,
@@ -12,6 +11,7 @@
     _get_flat_components,
     expand_compact_flow,
 )
+from pydantic import ValidationError
 
 
 class TestCompactNode:
@@ -134,7 +134,8 @@ def test_new_field_added_to_template(self):
 
     def test_template_key_copied(self):
         """Verify that expanding a node copies the template dict (shallow) so the
-        flat_components lookup table can be reused for another node of the same type."""
+        flat_components lookup table can be reused for another node of the same type.
+        """
         flat = {
             "MyComp": {"template": {"field": {"value": "original"}}},
         }
@@ -201,5 +202,5 @@ def test_multiple_nodes(self):
         assert len(result["nodes"]) == 2
 
     def test_invalid_compact_data_raises(self):
-        with pytest.raises(Exception):
+        with pytest.raises(ValidationError):
             expand_compact_flow({"nodes": "invalid"}, {})

src/backend/tests/unit/serialization/test_serialization_extended.py

@@ -1,36 +1,24 @@
 """Extended tests for the serialization module - covering internal helpers."""
 
-import math
 from datetime import datetime, timezone
 from decimal import Decimal
 from uuid import uuid4
 
 import numpy as np
 import pandas as pd
-import pytest
-from pydantic import BaseModel
-
 from langflow.serialization.serialization import (
-    UNSERIALIZABLE_SENTINEL,
-    _is_numpy_type,
     _serialize_bytes,
     _serialize_dataframe,
     _serialize_datetime,
-    _serialize_decimal,
-    _serialize_dict,
     _serialize_dispatcher,
-    _serialize_instance,
-    _serialize_iterator,
-    _serialize_list_tuple,
     _serialize_numpy_type,
     _serialize_pydantic,
-    _serialize_primitive,
     _serialize_series,
     _serialize_str,
-    _serialize_uuid,
     _truncate_value,
     serialize,
 )
+from pydantic import BaseModel
 
 
 class TestSerializeStrEdgeCases:
@@ -127,27 +115,6 @@ def test_float_inf(self):
         assert _serialize_dispatcher(float("inf"), None, None) is None
 
 
-class TestSerializeInstance:
-    """Tests for _serialize_instance."""
-
-    def test_custom_class(self):
-        class MyClass:
-            def __str__(self):
-                return "my_instance"
-
-        result = _serialize_instance(MyClass())
-        assert result == "my_instance"
-
-    def test_with_repr(self):
-        class MyClass:
-            def __repr__(self):
-                return "MyClass()"
-
-        obj = MyClass()
-        result = _serialize_instance(obj)
-        assert isinstance(result, str)
-
-
 class TestSerializeNumpyType:
     """Tests for numpy type serialization."""