diff --git a/tests/test_dtmf_aggregator.py b/tests/test_dtmf_aggregator.py index 40d3ece13..697d50d27 100644 --- a/tests/test_dtmf_aggregator.py +++ b/tests/test_dtmf_aggregator.py @@ -53,9 +53,9 @@ class TestDTMFAggregator(unittest.IsolatedAsyncioTestCase): frames_to_send = [ InputDTMFFrame(button=KeypadEntry.ONE), InputDTMFFrame(button=KeypadEntry.TWO), - SleepFrame(sleep=0.2), # This should trigger timeout + SleepFrame(), # This should trigger timeout InputDTMFFrame(button=KeypadEntry.THREE), - SleepFrame(sleep=0.2), # This should trigger another timeout + SleepFrame(), # This should trigger another timeout ] expected_down_frames = [ InputDTMFFrame, @@ -86,10 +86,10 @@ class TestDTMFAggregator(unittest.IsolatedAsyncioTestCase): InputDTMFFrame(button=KeypadEntry.ONE), InputDTMFFrame(button=KeypadEntry.TWO), InputDTMFFrame(button=KeypadEntry.POUND), # First sequence - SleepFrame(sleep=0.1), + SleepFrame(), InputDTMFFrame(button=KeypadEntry.FOUR), InputDTMFFrame(button=KeypadEntry.FIVE), - SleepFrame(sleep=0.3), # Second sequence via timeout + SleepFrame(), # Second sequence via timeout ] expected_down_frames = [ InputDTMFFrame, @@ -120,7 +120,7 @@ class TestDTMFAggregator(unittest.IsolatedAsyncioTestCase): frames_to_send = [ InputDTMFFrame(button=KeypadEntry.ONE), InputDTMFFrame(button=KeypadEntry.TWO), - SleepFrame(sleep=0.1), # Allow time for aggregation + SleepFrame(), # Allow time for aggregation EndFrame(), ] expected_down_frames = [ diff --git a/tests/test_stt_mute_filter.py b/tests/test_stt_mute_filter.py index 161962224..3d707dca7 100644 --- a/tests/test_stt_mute_filter.py +++ b/tests/test_stt_mute_filter.py @@ -124,14 +124,14 @@ class TestSTTMuteFilter(unittest.IsolatedAsyncioTestCase): frames_to_send = [ # Bot speaking - should mute BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), # Wait for StartedSpeaking to process + SleepFrame(), # Wait for StartedSpeaking to process InterimTranscriptionFrame( user_id="user1", text="This should be suppressed", timestamp="1234567890" ), TranscriptionFrame( user_id="user1", text="This should be suppressed", timestamp="1234567890" ), - SleepFrame(sleep=0.1), # Wait for transcription frames to queue + SleepFrame(), # Wait for transcription frames to queue BotStoppedSpeakingFrame(), # Bot not speaking - should pass through InterimTranscriptionFrame( diff --git a/tests/test_transcript_processor.py b/tests/test_transcript_processor.py index d446dc6d6..a2a057a13 100644 --- a/tests/test_transcript_processor.py +++ b/tests/test_transcript_processor.py @@ -129,13 +129,13 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): # Create test frames simulating bot speaking multiple text chunks frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), # Wait for StartedSpeaking to process + SleepFrame(), # Wait for StartedSpeaking to process TTSTextFrame(text="Hello"), TTSTextFrame(text="world!"), TTSTextFrame(text="How"), TTSTextFrame(text="are"), TTSTextFrame(text="you?"), - SleepFrame(sleep=0.1), # Wait for text frames to queue + SleepFrame(), # Wait for text frames to queue BotStoppedSpeakingFrame(), ] @@ -194,7 +194,7 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text=""), # Empty text TTSTextFrame(text=" "), # Just whitespace TTSTextFrame(text="\n"), # Just newline @@ -234,16 +234,16 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): # Simulate bot being interrupted mid-sentence frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text="Hello"), TTSTextFrame(text="world!"), - SleepFrame(sleep=0.1), + SleepFrame(), StartInterruptionFrame(), # User interrupts here - SleepFrame(sleep=0.1), + SleepFrame(), BotStartedSpeakingFrame(), TTSTextFrame(text="New"), TTSTextFrame(text="response"), - SleepFrame(sleep=0.1), + SleepFrame(), BotStoppedSpeakingFrame(), ] @@ -298,7 +298,7 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text="Hello"), TTSTextFrame(text="world"), # Pipeline ends here; run_test will automatically send EndFrame @@ -337,10 +337,10 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text="Hello"), TTSTextFrame(text="world"), - SleepFrame(sleep=0.1), # Ensure messages are processed + SleepFrame(), # Ensure messages are processed CancelFrame(), ] @@ -400,7 +400,7 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): # Test assistant processor frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text="Assistant"), TTSTextFrame(text="message"), BotStoppedSpeakingFrame(), @@ -440,7 +440,7 @@ class TestUserTranscriptProcessor(unittest.IsolatedAsyncioTestCase): # Test the specific pattern shared frames_to_send = [ BotStartedSpeakingFrame(), - SleepFrame(sleep=0.1), + SleepFrame(), TTSTextFrame(text="Hello"), TTSTextFrame(text=" there"), TTSTextFrame(text="!"),