refactor: move deep_merge_dicts and state_delta list tests to test_functions_parallel

Move tests for deep_merge_dicts and merge_parallel_function_response_events
with list state_delta merging from test_functions_simple.py to
test_functions_parallel.py per reviewer feedback.

Author: enjoykumawat

tests/unittests/flows/llm_flows/test_functions_parallel.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 
 from google.adk.agents.llm_agent import Agent
+from google.adk.events.event import Event
 from google.adk.events.event_actions import EventActions
+from google.adk.flows.llm_flows.functions import deep_merge_dicts
+from google.adk.flows.llm_flows.functions import merge_parallel_function_response_events
 from google.adk.tools.tool_context import ToolContext
 from google.genai import types
 import pytest
@@ -105,3 +108,83 @@ async def transfer_to_agent(
       },
       transfer_to_agent='test_sub_agent',
   )
+
+
+def test_deep_merge_dicts_concatenates_lists():
+  """Test that deep_merge_dicts concatenates list values instead of overwriting."""
+  d1 = {'state_delta': {'items': ['a']}}
+  d2 = {'state_delta': {'items': ['b']}}
+  result = deep_merge_dicts(d1, d2)
+  assert result['state_delta']['items'] == ['a', 'b']
+
+
+def test_deep_merge_dicts_overwrites_non_list_non_dict():
+  """Test that deep_merge_dicts still overwrites scalar values."""
+  d1 = {'key': 'old'}
+  d2 = {'key': 'new'}
+  result = deep_merge_dicts(d1, d2)
+  assert result['key'] == 'new'
+
+
+def test_deep_merge_dicts_merges_nested_dicts():
+  """Test that deep_merge_dicts recursively merges nested dicts."""
+  d1 = {'a': {'b': 1, 'c': 2}}
+  d2 = {'a': {'b': 3, 'd': 4}}
+  result = deep_merge_dicts(d1, d2)
+  assert result == {'a': {'b': 3, 'c': 2, 'd': 4}}
+
+
+def test_deep_merge_dicts_handles_mixed_list_and_non_list():
+  """Test that deep_merge_dicts overwrites when types differ (list vs non-list)."""
+  d1 = {'key': 'not_a_list'}
+  d2 = {'key': ['a', 'b']}
+  result = deep_merge_dicts(d1, d2)
+  assert result['key'] == ['a', 'b']
+
+  d1 = {'key': ['a', 'b']}
+  d2 = {'key': 'not_a_list'}
+  result = deep_merge_dicts(d1, d2)
+  assert result['key'] == 'not_a_list'
+
+
+def test_merge_parallel_function_response_events_merges_state_delta_lists():
+  """Test that parallel events with list state_delta values are concatenated, not overwritten."""
+  invocation_id = 'base_invocation_123'
+
+  event1 = Event(
+      invocation_id=invocation_id,
+      author='tool',
+      content=types.Content(
+          role='user',
+          parts=[
+              types.Part(
+                  function_response=types.FunctionResponse(
+                      name='func_1',
+                      response={'result': 'ok'},
+                  )
+              )
+          ],
+      ),
+      actions=EventActions(state_delta={'items': ['a']}),
+  )
+
+  event2 = Event(
+      invocation_id=invocation_id,
+      author='tool',
+      content=types.Content(
+          role='user',
+          parts=[
+              types.Part(
+                  function_response=types.FunctionResponse(
+                      name='func_2',
+                      response={'result': 'ok'},
+                  )
+              )
+          ],
+      ),
+      actions=EventActions(state_delta={'items': ['b']}),
+  )
+
+  merged_event = merge_parallel_function_response_events([event1, event2])
+
+  assert merged_event.actions.state_delta == {'items': ['a', 'b']}

tests/unittests/flows/llm_flows/test_functions_simple.py

@@ -24,7 +24,6 @@
 from google.adk.events.event import Event
 from google.adk.events.event_actions import EventActions
 from google.adk.events.ui_widget import UiWidget
-from google.adk.flows.llm_flows.functions import deep_merge_dicts
 from google.adk.flows.llm_flows.functions import find_matching_function_call
 from google.adk.flows.llm_flows.functions import handle_function_calls_async
 from google.adk.flows.llm_flows.functions import handle_function_calls_live
@@ -1865,73 +1864,3 @@ async def slow_fn_2() -> dict[str, str]:
 
   await asyncio.sleep(0)
   assert len(invocation_context.active_non_blocking_tool_tasks) == 0
-
-
-def test_deep_merge_dicts_concatenates_lists():
-  """Test that deep_merge_dicts concatenates list values instead of overwriting."""
-  d1 = {"state_delta": {"items": ["a"]}}
-  d2 = {"state_delta": {"items": ["b"]}}
-  result = deep_merge_dicts(d1, d2)
-  assert result["state_delta"]["items"] == ["a", "b"]
-
-
-def test_deep_merge_dicts_overwrites_non_list_non_dict():
-  """Test that deep_merge_dicts still overwrites scalar values."""
-  d1 = {"key": "old"}
-  d2 = {"key": "new"}
-  result = deep_merge_dicts(d1, d2)
-  assert result["key"] == "new"
-
-
-def test_deep_merge_dicts_merges_nested_dicts():
-  """Test that deep_merge_dicts recursively merges nested dicts."""
-  d1 = {"a": {"b": 1, "c": 2}}
-  d2 = {"a": {"b": 3, "d": 4}}
-  result = deep_merge_dicts(d1, d2)
-  assert result == {"a": {"b": 3, "c": 2, "d": 4}}
-
-
-def test_deep_merge_dicts_handles_mixed_list_and_non_list():
-  """Test that deep_merge_dicts overwrites when types differ (list vs non-list)."""
-  d1 = {"key": "not_a_list"}
-  d2 = {"key": ["a", "b"]}
-  result = deep_merge_dicts(d1, d2)
-  assert result["key"] == ["a", "b"]
-
-  d1 = {"key": ["a", "b"]}
-  d2 = {"key": "not_a_list"}
-  result = deep_merge_dicts(d1, d2)
-  assert result["key"] == "not_a_list"
-
-
-def test_merge_parallel_function_response_events_merges_state_delta_lists():
-  """Test that parallel events with list state_delta values are concatenated, not overwritten."""
-  invocation_id = "base_invocation_123"
-
-  event1 = Event(
-      invocation_id=invocation_id,
-      author="tool",
-      content=types.Content(
-          role="user",
-          parts=[types.Part(function_response=types.FunctionResponse(
-              name="func_1", response={"result": "ok"},
-          ))],
-      ),
-      actions=EventActions(state_delta={"items": ["a"]}),
-  )
-
-  event2 = Event(
-      invocation_id=invocation_id,
-      author="tool",
-      content=types.Content(
-          role="user",
-          parts=[types.Part(function_response=types.FunctionResponse(
-              name="func_2", response={"result": "ok"},
-          ))],
-      ),
-      actions=EventActions(state_delta={"items": ["b"]}),
-  )
-
-  merged_event = merge_parallel_function_response_events([event1, event2])
-
-  assert merged_event.actions.state_delta == {"items": ["a", "b"]}