style: fix style

test/unit/agents/bdi/behaviours/test_belief_setter.py

@@ -1,6 +1,6 @@
 import json
 import logging
-from unittest.mock import MagicMock, AsyncMock, call
+from unittest.mock import AsyncMock, MagicMock, call
 
 import pytest
 
@@ -26,11 +26,11 @@ def belief_setter(mock_agent, mocker):
     # Patch the settings to use a predictable agent name
     mocker.patch(
         "control_backend.agents.bdi.behaviours.belief_setter.settings.agent_settings.belief_collector_agent_name",
-        COLLECTOR_AGENT_NAME
+        COLLECTOR_AGENT_NAME,
     )
     # Patch asyncio.sleep to prevent tests from actually waiting
     mocker.patch("asyncio.sleep", return_value=None)
-    
+
     setter = BeliefSetter()
     setter.agent = mock_agent
     # Mock the receive method, we will control its return value in each test
@@ -69,7 +69,7 @@ async def test_run_message_received(belief_setter, mocker):
     Test that when a message is received, _process_message is called.
     """
     # Arrange
-    msg = MagicMock();
+    msg = MagicMock()
     belief_setter.receive.return_value = msg
     mocker.patch.object(belief_setter, "_process_message")
 
@@ -115,14 +115,9 @@ def test_process_belief_message_valid_json(belief_setter, mocker):
     Test processing a valid belief message with correct thread and JSON body.
     """
     # Arrange
-    beliefs_payload = {
-        "is_hot": [["kitchen"]],
-        "is_clean": [["kitchen"], ["bathroom"]]
-    }
+    beliefs_payload = {"is_hot": [["kitchen"]], "is_clean": [["kitchen"], ["bathroom"]]}
     msg = create_mock_message(
-        sender_node=COLLECTOR_AGENT_JID,
-        body=json.dumps(beliefs_payload),
-        thread="beliefs"
+        sender_node=COLLECTOR_AGENT_JID, body=json.dumps(beliefs_payload), thread="beliefs"
     )
     mock_set_beliefs = mocker.patch.object(belief_setter, "_set_beliefs")
 
@@ -139,9 +134,7 @@ def test_process_belief_message_invalid_json(belief_setter, mocker, caplog):
     """
     # Arrange
     msg = create_mock_message(
-        sender_node=COLLECTOR_AGENT_JID,
-        body="this is not a json string",
-        thread="beliefs"
+        sender_node=COLLECTOR_AGENT_JID, body="this is not a json string", thread="beliefs"
     )
     mock_set_beliefs = mocker.patch.object(belief_setter, "_set_beliefs")
 
@@ -160,9 +153,7 @@ def test_process_belief_message_wrong_thread(belief_setter, mocker):
     """
     # Arrange
     msg = create_mock_message(
-        sender_node=COLLECTOR_AGENT_JID,
-        body='{"some": "data"}',
-        thread="not_beliefs"
+        sender_node=COLLECTOR_AGENT_JID, body='{"some": "data"}', thread="not_beliefs"
     )
     mock_set_beliefs = mocker.patch.object(belief_setter, "_set_beliefs")
 
@@ -172,16 +163,13 @@ def test_process_belief_message_wrong_thread(belief_setter, mocker):
     # Assert
     mock_set_beliefs.assert_not_called()
 
+
 def test_process_belief_message_empty_body(belief_setter, mocker):
     """
     Test that a message with an empty body is ignored.
     """
     # Arrange
-    msg = create_mock_message(
-        sender_node=COLLECTOR_AGENT_JID,
-        body="",
-        thread="beliefs"
-    )
+    msg = create_mock_message(sender_node=COLLECTOR_AGENT_JID, body="", thread="beliefs")
     mock_set_beliefs = mocker.patch.object(belief_setter, "_set_beliefs")
 
     # Act
@@ -198,9 +186,9 @@ def test_set_beliefs_success(belief_setter, mock_agent, caplog):
     # Arrange
     beliefs_to_set = {
         "is_hot": [["kitchen"], ["living_room"]],
-        "door_is": [["front_door", "closed"]]
+        "door_is": [["front_door", "closed"]],
     }
-    
+
     # Act
     with caplog.at_level(logging.INFO):
         belief_setter._set_beliefs(beliefs_to_set)
@@ -209,11 +197,11 @@ def test_set_beliefs_success(belief_setter, mock_agent, caplog):
     expected_calls = [
         call("is_hot", "kitchen"),
         call("is_hot", "living_room"),
-        call("door_is", "front_door", "closed")
+        call("door_is", "front_door", "closed"),
     ]
     mock_agent.bdi.set_belief.assert_has_calls(expected_calls, any_order=True)
     assert mock_agent.bdi.set_belief.call_count == 3
-    
+
     # Check logs
     assert "Set belief is_hot with arguments ['kitchen']" in caplog.text
     assert "Set belief is_hot with arguments ['living_room']" in caplog.text