Update test_tools.py

BlueBrain · Nov 14, 2024 · dd5a926 · dd5a926
1 parent ad8cb0d
commit dd5a926
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/tests/app/database/test_tools.py b/tests/app/database/test_tools.py
@@ -53,13 +53,16 @@ async def test_get_tool_calls(
         message_id = messages[-1]["message_id"]
         tool_calls = app_client.get(f"/tools/{thread_id}/{message_id}").json()
 
-    assert tool_calls[0] == {
-        "call_id": "call_zHhwfNLSvGGHXMoILdIYtDVI",
-        "name": "get-morpho-tool",
-        "arguments": {
-            "brain_region_id": "http://api.brain-map.org/api/v2/data/Structure/549"
-        },
-    }
+    assert (
+        tool_calls[0]
+        == ToolCallSchema(
+            call_id="call_zHhwfNLSvGGHXMoILdIYtDVI",
+            name="get-morpho-tool",
+            arguments={
+                "brain_region_id": "http://api.brain-map.org/api/v2/data/Structure/549"
+            },
+        ).model_dump()
+    )
 
 
 @pytest.mark.httpx_mock(can_send_already_matched_responses=True)