Travis Vasceannie
3acb779569
- Deleted .env.example file as it is no longer needed. - Added .gitignore to manage ignored files and directories. - Introduced CLAUDE.md for AI provider integration documentation. - Created dev.sh for development setup and scripts. - Updated Dockerfile and Dockerfile.production for improved build processes. - Added multiple test files and directories for comprehensive testing. - Introduced new utility and service files for enhanced functionality. - Organized codebase with new directories and files for better maintainability.
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
"""
|
|
Test fixtures and utilities for NVIDIA NeMo speaker diarization testing.
|
|
|
|
This module provides comprehensive testing infrastructure including:
|
|
- Mock NeMo models and services
|
|
- Audio sample generation
|
|
- Test data management
|
|
- Performance testing utilities
|
|
"""
|
|
|
|
from .audio_samples import (BASIC_SCENARIOS, CHALLENGING_SCENARIOS,
|
|
TEST_SCENARIOS, AudioFileManager,
|
|
AudioSampleGenerator, AudioScenario,
|
|
TestDataGenerator, create_quick_test_files,
|
|
get_scenario_by_difficulty)
|
|
# Import existing Discord mocks for compatibility
|
|
from .mock_discord import (MockAudioSource, MockBot, MockContext,
|
|
MockDiscordMember, MockDiscordUser, MockGuild,
|
|
MockInteraction, MockInteractionFollowup,
|
|
MockInteractionResponse, MockMessage,
|
|
MockPermissions, MockTextChannel, MockVoiceChannel,
|
|
MockVoiceClient, MockVoiceState,
|
|
create_mock_voice_scenario)
|
|
from .nemo_mocks import (MockAudioGenerator, MockDiarizationResultGenerator,
|
|
MockMarbleNetVAD, MockMSDDNeuralDiarizer,
|
|
MockNeMoCascadedModels, MockNeMoModelFactory,
|
|
MockNeMoSortformerModel, MockServiceResponses,
|
|
MockTitaNetSpeaker, cleanup_mock_files,
|
|
create_mock_nemo_environment, generate_test_manifest,
|
|
generate_test_rttm_content, patch_nemo_models)
|
|
|
|
__all__ = [
|
|
# NeMo Mock Classes
|
|
"MockNeMoSortformerModel",
|
|
"MockNeMoCascadedModels",
|
|
"MockMarbleNetVAD",
|
|
"MockTitaNetSpeaker",
|
|
"MockMSDDNeuralDiarizer",
|
|
"MockNeMoModelFactory",
|
|
"MockAudioGenerator",
|
|
"MockDiarizationResultGenerator",
|
|
"MockServiceResponses",
|
|
# NeMo Mock Functions
|
|
"patch_nemo_models",
|
|
"create_mock_nemo_environment",
|
|
"generate_test_manifest",
|
|
"generate_test_rttm_content",
|
|
"cleanup_mock_files",
|
|
# Audio Sample Classes
|
|
"AudioScenario",
|
|
"AudioSampleGenerator",
|
|
"AudioFileManager",
|
|
"TestDataGenerator",
|
|
# Audio Sample Functions and Constants
|
|
"TEST_SCENARIOS",
|
|
"CHALLENGING_SCENARIOS",
|
|
"BASIC_SCENARIOS",
|
|
"get_scenario_by_difficulty",
|
|
"create_quick_test_files",
|
|
# Discord Mock Classes
|
|
"MockAudioSource",
|
|
"MockBot",
|
|
"MockContext",
|
|
"MockDiscordMember",
|
|
"MockDiscordUser",
|
|
"MockGuild",
|
|
"MockInteraction",
|
|
"MockInteractionFollowup",
|
|
"MockInteractionResponse",
|
|
"MockMessage",
|
|
"MockPermissions",
|
|
"MockTextChannel",
|
|
"MockVoiceChannel",
|
|
"MockVoiceClient",
|
|
"MockVoiceState",
|
|
"create_mock_voice_scenario",
|
|
]
|