diff options
Diffstat (limited to 'app/tests/test_converter_progress.py')
| -rw-r--r-- | app/tests/test_converter_progress.py | 31 |
1 files changed, 19 insertions, 12 deletions
diff --git a/app/tests/test_converter_progress.py b/app/tests/test_converter_progress.py index 0f708a1..00bcc46 100644 --- a/app/tests/test_converter_progress.py +++ b/app/tests/test_converter_progress.py @@ -14,7 +14,14 @@ from contextlib import redirect_stdout from pathlib import Path from unittest.mock import MagicMock, patch -from converter import config, tts +from converter import config +from converter.clients import ( + BACKEND_AUDIOCPP, + BACKEND_FASTER, + BACKEND_QWEN, + VOICE_MODE_CLONE, + VOICE_MODE_CUSTOM, +) from converter import converter as converter_mod from converter.converter import ( AudiobookConverter, @@ -26,24 +33,24 @@ from converter.converter import ( class VoiceModeForTests(unittest.TestCase): def test_faster_always_clones(self): - self.assertEqual(voice_mode_for(tts.BACKEND_FASTER), - tts.VOICE_MODE_CLONE) + self.assertEqual(voice_mode_for(BACKEND_FASTER), + VOICE_MODE_CLONE) def test_audiocpp_voice_clones(self): - self.assertEqual(voice_mode_for(tts.BACKEND_AUDIOCPP, voice="narrator"), - tts.VOICE_MODE_CLONE) + self.assertEqual(voice_mode_for(BACKEND_AUDIOCPP, voice="narrator"), + VOICE_MODE_CLONE) def test_audiocpp_no_voice_is_custom(self): - self.assertEqual(voice_mode_for(tts.BACKEND_AUDIOCPP), - tts.VOICE_MODE_CUSTOM) + self.assertEqual(voice_mode_for(BACKEND_AUDIOCPP), + VOICE_MODE_CUSTOM) def test_qwen_clone_wav_clones(self): - self.assertEqual(voice_mode_for(tts.BACKEND_QWEN, clone="x.wav"), - tts.VOICE_MODE_CLONE) + self.assertEqual(voice_mode_for(BACKEND_QWEN, clone="x.wav"), + VOICE_MODE_CLONE) def test_qwen_no_clone_is_custom(self): - self.assertEqual(voice_mode_for(tts.BACKEND_QWEN), - tts.VOICE_MODE_CUSTOM) + self.assertEqual(voice_mode_for(BACKEND_QWEN), + VOICE_MODE_CUSTOM) class PromptOverwriteConfirmTests(unittest.TestCase): @@ -96,7 +103,7 @@ class _ConvertFixture: with patch.object(converter_mod, "QwenTTSClient", return_value=MagicMock()): converter = AudiobookConverter( - voice_mode=tts.VOICE_MODE_CUSTOM, backend=tts.BACKEND_QWEN, + voice_mode=VOICE_MODE_CUSTOM, backend=BACKEND_QWEN, output_format="mp3", language="English", progress=progress, cancel=cancel) converter.tts.process_chunk_with_retry.return_value = "chunk_0001.wav" |