mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 21:50:28 +02:00
Artem Akymenko
8ccdc85ccb
- Rename abogen/webui/routes/utils/preview.py → synthesize.py The file contains the core TTS synthesis pipeline (generate_preview_audio, synthesize_preview), not just preview logic. Name now matches responsibility. - Remove dead code from voice.py: get_preview_pipeline(), synthesize_audio_from_normalized(), _preview_pipeline_lock, _preview_pipelines, and unused imports (threading, numpy, create_pipeline, get_new_voice, _select_device, _to_float32, SAMPLE_RATE, SPLIT_PATTERN). These were never called — identical logic lives in synthesize.py. - Update imports in api.py, voices.py, and test_preview_applies_manual_overrides.py - 7 new tests in test_synthesize_module.py enforce file naming and import rules - 7 tests in test_domain_imports.py updated for renamed module
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from abogen.webui.routes.utils import synthesize
|
|
|
|
|
|
def test_preview_applies_manual_override_before_normalization(monkeypatch):
|
|
# Don't run real TTS/normalization; just exercise the override stage by
|
|
# forcing provider=kokoro and then stubbing normalize_for_pipeline.
|
|
|
|
monkeypatch.setattr(synthesize, "get_preview_pipeline", lambda language, device: None)
|
|
|
|
# Stub normalize_for_pipeline to be identity; we only care that overrides run.
|
|
class _Norm:
|
|
@staticmethod
|
|
def normalize_for_pipeline(text):
|
|
return text
|
|
|
|
monkeypatch.setitem(
|
|
__import__("sys").modules, "abogen.kokoro_text_normalization", _Norm
|
|
)
|
|
|
|
# And stub the kokoro pipeline path so generate_preview_audio won't proceed.
|
|
# We'll instead validate by calling the override logic through generate_preview_audio
|
|
# with provider=supertonic and stub create_backend to capture input.
|
|
captured = {}
|
|
|
|
class DummyPipeline:
|
|
def __init__(self, **kwargs):
|
|
pass
|
|
|
|
def __call__(self, text, **kwargs):
|
|
captured["text"] = text
|
|
return iter(())
|
|
|
|
from abogen.tts_plugin import utils
|
|
|
|
original_create_pipeline = utils.create_pipeline
|
|
|
|
def _mock_create_pipeline(backend_id, **kwargs):
|
|
if backend_id == "supertonic":
|
|
return DummyPipeline(**kwargs)
|
|
return original_create_pipeline(backend_id, **kwargs)
|
|
|
|
monkeypatch.setattr(utils, "create_pipeline", _mock_create_pipeline)
|
|
|
|
try:
|
|
synthesize.generate_preview_audio(
|
|
text="He said Unfu*k loudly.",
|
|
voice_spec="M1",
|
|
language="en",
|
|
speed=1.0,
|
|
use_gpu=False,
|
|
tts_provider="supertonic",
|
|
manual_overrides=[{"token": "Unfu*k", "pronunciation": "Unfuck"}],
|
|
)
|
|
except Exception:
|
|
# generate_preview_audio will raise because no audio chunks; that's fine.
|
|
pass
|
|
|
|
assert "text" in captured
|
|
assert "Unfuck" in captured["text"]
|
|
assert "Unfu*k" not in captured["text"]
|