aboutsummaryrefslogtreecommitdiff
path: root/app/tests/test_converter_progress.py
diff options
context:
space:
mode:
authorhistoria <historiavg@proton.me>2026-08-26 01:43:41 -0400
committerhistoria <historiavg@proton.me>2026-08-26 01:43:41 -0400
commitacbd9ff2c91182d96c57ffb57bee6e9b3fcbcbd4 (patch)
treee336c11f2a57cff5566e249aa5d5477a3dc63c55 /app/tests/test_converter_progress.py
parent104a0d65c1ba37847c15b64212b7fec8ba371ccb (diff)
downloadtts-audiobook-generator-acbd9ff2c91182d96c57ffb57bee6e9b3fcbcbd4.tar.gz
refactor: split tts.py into per-backend packages
Diffstat (limited to 'app/tests/test_converter_progress.py')
-rw-r--r--app/tests/test_converter_progress.py31
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"