public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-04-13 09:39:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixed final test

Browse source
This commit is contained in:
Fireblade 2025-02-15 09:49:15 -05:00
parent 1a6e7abac3
commit 8c457c3292
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
api/tests/test_openai_endpoints.py
View file

@ -258,7 +258,7 @@ def test_openai_speech_endpoint(
"""Test the OpenAI-compatible speech endpoint with basic MP3 generation""" """Test the OpenAI-compatible speech endpoint with basic MP3 generation"""
# Configure mocks # Configure mocks
mock_tts_service.generate_audio.return_value = (np.zeros(1000), AudioChunk(np.zeros(1000,np.int16))) 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( response = client.post(
"/v1/audio/speech", "/v1/audio/speech",
@ -273,10 +273,10 @@ def test_openai_speech_endpoint(
assert response.status_code == 200 assert response.status_code == 200
assert response.headers["content-type"] == "audio/mpeg" assert response.headers["content-type"] == "audio/mpeg"
assert len(response.content) > 0 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_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): def test_openai_speech_streaming(mock_tts_service, test_voice, mock_audio_bytes):