test/integration/agents/perception/vad_agent/test_vad_with_audio.py

@@ -87,7 +87,7 @@ async def test_real_audio(mocker):
             return None
 
     vad_agent.audio_in_poller = DummyPoller(chunk_bytes, vad_agent)
-    vad_agent._ready = True
+    vad_agent._ready = AsyncMock()
     vad_agent._running = True
     vad_agent.i_since_speech = 0
 

test/unit/agents/bdi/test_belief_collector.py

@@ -3,7 +3,7 @@ from unittest.mock import AsyncMock
 
 import pytest
 
-from control_backend.agents.bdi.belief_collector_agent.belief_collector_agent import (
+from control_backend.agents.bdi import (
     BDIBeliefCollectorAgent,
 )
 from control_backend.core.agent_system import InternalMessage

test/unit/agents/bdi/test_text_extractor.py

@@ -3,7 +3,7 @@ from unittest.mock import AsyncMock
 
 import pytest
 
-from control_backend.agents.bdi.text_belief_extractor_agent.text_belief_extractor_agent import (
+from control_backend.agents.bdi import (
     TextBeliefExtractorAgent,
 )
 from control_backend.core.agent_system import InternalMessage

test/unit/agents/perception/vad_agent/test_vad_streaming.py

@@ -61,7 +61,7 @@ async def simulate_streaming_with_probabilities(streaming, probabilities: list[f
             return None
 
     streaming.audio_in_poller = DummyPoller(chunks, streaming)
-    streaming._ready = True
+    streaming._ready = AsyncMock()
     streaming._running = True
 
     await streaming._streaming_loop()
@@ -116,7 +116,7 @@ async def test_no_data(audio_out_socket, vad_agent):
 
     vad_agent.audio_out_socket = audio_out_socket
     vad_agent.audio_in_poller = DummyPoller()
-    vad_agent._ready = True
+    vad_agent._ready = AsyncMock()
     vad_agent._running = True
 
     await vad_agent._streaming_loop()