mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 05:40:26 +02:00
feat: Add _chunk_text_for_tts function to prioritize raw text for TTS synthesis and implement related tests
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
from abogen.web.conversion_runner import _chunk_text_for_tts
|
||||
|
||||
|
||||
def test_chunk_text_for_tts_prefers_text_over_normalized_text():
|
||||
entry = {
|
||||
# Simulate a pre-normalized chunk that lost the asterisk.
|
||||
"normalized_text": "Unfuk",
|
||||
# Raw chunk should preserve censored token for manual overrides.
|
||||
"text": "Unfu*k",
|
||||
}
|
||||
|
||||
assert _chunk_text_for_tts(entry) == "Unfu*k"
|
||||
|
||||
|
||||
def test_chunk_text_for_tts_falls_back_to_original_text_then_normalized_text():
|
||||
entry = {
|
||||
"original_text": "Hello * world",
|
||||
"normalized_text": "Hello world",
|
||||
}
|
||||
assert _chunk_text_for_tts(entry) == "Hello * world"
|
||||
|
||||
entry2 = {
|
||||
"normalized_text": "Only normalized",
|
||||
}
|
||||
assert _chunk_text_for_tts(entry2) == "Only normalized"
|
||||
Reference in New Issue
Block a user