✅ (test_schema.py): add additional test cases for post_process_type f…

…unction to cover various Union types and combinations for better test coverage and accuracy

src/backend/tests/unit/test_schema.py

@@ -80,6 +80,25 @@ def test_post_process_type_function(self):
         assert set(post_process_type(dict)) == {dict}
         assert set(post_process_type(tuple)) == {tuple}
 
+        # Union with custom types
+        assert set(post_process_type(Union[Data, str])) == {Data, str}
+        assert set(post_process_type(Data | str)) == {Data, str}
+        assert set(post_process_type(Data | int | list[str])) == {Data, int, str}
+
+        # More complex combinations with Data
+        assert set(post_process_type(Data | list[float])) == {Data, float}
+        assert set(post_process_type(Data | Union[int, str])) == {Data, int, str}
+        assert set(post_process_type(Data | list[int] | None)) == {Data, int, type(None)}
+        assert set(post_process_type(Data | Union[float, None])) == {Data, float, type(None)}
+
+        # Multiple Data types combined
+        assert set(post_process_type(Union[Data, Union[str, float]])) == {Data, str, float}
+        assert set(post_process_type(Union[Data | float | str, int])) == {Data, int, float, str}
+
+        # Testing with nested unions and lists
+        assert set(post_process_type(Union[list[Data], list[Union[int, str]]])) == {Data, int, str}
+        assert set(post_process_type(Data | list[Union[float, str]])) == {Data, float, str}
+
     def test_input_to_dict(self):
         input_obj = Input(field_type="str")
         assert input_obj.to_dict() == {