Spaces:
Runtime error
Runtime error
File size: 2,237 Bytes
05b45a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
from pathlib import Path
from unittest.mock import AsyncMock, MagicMock, patch
import numpy as np
import pytest
import pytest_asyncio
import torch
from api.src.inference.model_manager import ModelManager
from api.src.inference.voice_manager import VoiceManager
from api.src.services.tts_service import TTSService
from api.src.structures.model_schemas import VoiceConfig
@pytest.fixture
def mock_voice_tensor():
"""Load a real voice tensor for testing."""
voice_path = os.path.join(
os.path.dirname(os.path.dirname(__file__)), "src/voices/af_bella.pt"
)
return torch.load(voice_path, map_location="cpu", weights_only=False)
@pytest.fixture
def mock_audio_output():
"""Load pre-generated test audio for consistent testing."""
test_audio_path = os.path.join(
os.path.dirname(__file__), "test_data/test_audio.npy"
)
return np.load(test_audio_path) # Return as numpy array instead of bytes
@pytest_asyncio.fixture
async def mock_model_manager(mock_audio_output):
"""Mock model manager for testing."""
manager = AsyncMock(spec=ModelManager)
manager.get_backend = MagicMock()
async def mock_generate(*args, **kwargs):
# Simulate successful audio generation
return np.random.rand(24000).astype(np.float32) # 1 second of random audio data
manager.generate = AsyncMock(side_effect=mock_generate)
return manager
@pytest_asyncio.fixture
async def mock_voice_manager(mock_voice_tensor):
"""Mock voice manager for testing."""
manager = AsyncMock(spec=VoiceManager)
manager.get_voice_path = MagicMock(return_value="/mock/path/voice.pt")
manager.load_voice = AsyncMock(return_value=mock_voice_tensor)
manager.list_voices = AsyncMock(return_value=["voice1", "voice2"])
manager.combine_voices = AsyncMock(return_value="voice1_voice2")
return manager
@pytest_asyncio.fixture
async def tts_service(mock_model_manager, mock_voice_manager):
"""Get mocked TTS service instance."""
service = TTSService()
service.model_manager = mock_model_manager
service._voice_manager = mock_voice_manager
return service
@pytest.fixture
def test_voice():
"""Return a test voice name."""
return "voice1"
|