mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 21:50:28 +02:00
26 lines
773 B
Python
26 lines
773 B
Python
from abogen.webui.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"
|