Update test_agent.py with comments

Author: Unshure

tests/strands/agent/test_agent.py

@@ -1332,32 +1332,41 @@ def test_non_json_serializable_state_throws_error():
 def test_agent_state_breaks_dict_reference():
     ref_dict = {"hello": "world"}
     agent = Agent(state=ref_dict)
+
+    # Make sure shallow object references do not affect state maintained by AgentState
     ref_dict["hello"] = object()
 
+    # This will fail if AgentState reflects the updated reference
     json.dumps(agent.state.get())
 
 
 def test_agent_state_breaks_deep_dict_reference():
     ref_dict = {"world": "!"}
     init_dict = {"hello": ref_dict}
     agent = Agent(state=init_dict)
+    # Make sure deep reference changes do not affect state mained by AgentState
     ref_dict["world"] = object()
 
+    # This will fail if AgentState reflects the updated reference
     json.dumps(agent.state.get())
 
 
 def test_agent_state_set_breaks_dict_reference():
     agent = Agent()
     ref_dict = {"hello": "world"}
+    # Set should copy the input, and not maintain the reference to the original object
     agent.state.set("hello", ref_dict)
     ref_dict["hello"] = object()
 
+    # This will fail if AgentState reflects the updated reference
     json.dumps(agent.state.get())
 
 
 def test_agent_state_get_breaks_deep_dict_reference():
     agent = Agent(state={"hello": {"world": "!"}})
+    # Get should not return a reference to the internal state
     ref_state = agent.state.get()
     ref_state["hello"]["world"] = object()
 
+    # This will fail if AgentState reflects the updated reference
     json.dumps(agent.state.get())