From 8c457c3292f5c1b2828fda334ec98d3fa648f220 Mon Sep 17 00:00:00 2001 From: Fireblade Date: Sat, 15 Feb 2025 09:49:15 -0500 Subject: [PATCH] fixed final test --- api/tests/test_openai_endpoints.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/tests/test_openai_endpoints.py b/api/tests/test_openai_endpoints.py index b02cb5f..82873e1 100644 --- a/api/tests/test_openai_endpoints.py +++ b/api/tests/test_openai_endpoints.py @@ -258,7 +258,7 @@ def test_openai_speech_endpoint( """Test the OpenAI-compatible speech endpoint with basic MP3 generation""" # Configure mocks mock_tts_service.generate_audio.return_value = (np.zeros(1000), AudioChunk(np.zeros(1000,np.int16))) - mock_convert.return_value = mock_audio_bytes + mock_convert.return_value = (mock_audio_bytes,AudioChunk(np.zeros(1000,np.int16))) response = client.post( "/v1/audio/speech", @@ -273,10 +273,10 @@ def test_openai_speech_endpoint( assert response.status_code == 200 assert response.headers["content-type"] == "audio/mpeg" assert len(response.content) > 0 - assert response.content == mock_audio_bytes + assert response.content == mock_audio_bytes + mock_audio_bytes mock_tts_service.generate_audio.assert_called_once() - mock_convert.assert_called_once() + assert mock_convert.call_count == 2 def test_openai_speech_streaming(mock_tts_service, test_voice, mock_audio_bytes):