Fixing unit tests

singnet · Jul 22, 2024 · 9c7f82f · 9c7f82f
1 parent c75723a
commit 9c7f82f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -316,7 +316,10 @@ def test_count_atoms_success(self, mock_request, client):
 
     def test_get_atoms_by_field(self, mock_request, client):
         query = [{'field': 'name', 'value': 'test'}]
-        expected_request_data = {"action": "get_atoms_by_field", "input": {'query': query}}
+        expected_request_data = {
+            "action": "get_atoms_by_field",
+            "input": {'query': {k['field']: k['value'] for k in query}},
+        }
         expected_response = (14, 26)
         mock_request.return_value.status_code = 200
         mock_request.return_value.content = serialize(expected_response)