Jatin K MalikandGitHub
3645c692fd
* Add GitHub Actions CI/CD workflows and status badges * Fix CI workflow dependency installation * Fix CI workflow: remove pip caching and add explicit requirements.txt * Fix CI workflow: add system dependencies for PyGObject and pycairo * Refactor code for improved readability and consistency - Cleaned up whitespace and formatting in audio_feedback.py, config_manager.py, keyboard_shortcuts.py, tray_indicator.py, run_basic_tests.py, and test_basic.py. - Enhanced the structure of the code by aligning comments and ensuring consistent indentation. - Updated function definitions and calls to maintain a uniform style across the codebase. - Improved logging messages for better clarity during debugging. - Ensured all files end with a newline for better compatibility with various tools. * Refactor imports for consistency and clarity across multiple modules * Fix test imports by setting PYTHONPATH and adding conftest.py * Update CI workflow to use Python 3.12 and ensure proper import path in tests * Fix CI: Add GObject Introspection development dependencies * Update CI: Use virtual environment like local setup * Fix CI * Update README and setup.py: Change Python version to 3.8 and correct author email * Restructure tests for better organization and maintainability * Fix formatting * Update CI workflow to remove redundant test step * Fix CI tests by properly mocking VOSK model loading * Update CI and Tests * Simplify
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
"""
|
|
Tests for the SpeechRecognitionManager component.
|
|
"""
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from src.speech_recognition.recognition_manager import (
|
|
RecognitionState,
|
|
SpeechRecognitionManager,
|
|
)
|
|
|
|
|
|
class TestSpeechRecognition(unittest.TestCase):
|
|
"""Test cases for the speech recognition manager."""
|
|
|
|
@patch("src.speech_recognition.recognition_manager.os.path.exists")
|
|
@patch("src.speech_recognition.recognition_manager.os.makedirs")
|
|
@patch(
|
|
"src.speech_recognition.recognition_manager.SpeechRecognitionManager._get_vosk_model_path"
|
|
)
|
|
def test_init_state(self, mock_get_model_path, mock_makedirs, mock_exists):
|
|
"""Test the initial state of the speech recognition manager."""
|
|
# Mock the file existence check and model path
|
|
mock_exists.return_value = True
|
|
mock_get_model_path.return_value = "/fake/path/to/model"
|
|
|
|
# We need to patch the import of vosk to avoid errors
|
|
with patch.dict("sys.modules", {"vosk": MagicMock()}):
|
|
manager = SpeechRecognitionManager(engine="vosk", model_size="small")
|
|
assert manager.state == RecognitionState.IDLE
|
|
assert manager.engine == "vosk"
|
|
assert manager.model_size == "small"
|
|
|
|
@patch("src.speech_recognition.recognition_manager.os.path.exists")
|
|
@patch("src.speech_recognition.recognition_manager.os.makedirs")
|
|
@patch(
|
|
"src.speech_recognition.recognition_manager.SpeechRecognitionManager._get_vosk_model_path"
|
|
)
|
|
def test_callbacks(self, mock_get_model_path, mock_makedirs, mock_exists):
|
|
"""Test callback registration and invocation."""
|
|
# Mock the file existence check and model path
|
|
mock_exists.return_value = True
|
|
mock_get_model_path.return_value = "/fake/path/to/model"
|
|
|
|
# We need to patch the import of vosk to avoid errors
|
|
with patch.dict("sys.modules", {"vosk": MagicMock()}):
|
|
manager = SpeechRecognitionManager(engine="vosk", model_size="small")
|
|
|
|
# Mock callbacks
|
|
text_callback = MagicMock()
|
|
state_callback = MagicMock()
|
|
|
|
# Register callbacks
|
|
manager.register_text_callback(text_callback)
|
|
manager.register_state_callback(state_callback)
|
|
|
|
# Verify callbacks were registered
|
|
assert text_callback in manager.text_callbacks
|
|
assert state_callback in manager.state_callbacks
|
|
|
|
# Test state update
|
|
manager._update_state(RecognitionState.LISTENING)
|
|
state_callback.assert_called_once_with(RecognitionState.LISTENING)
|