package com.aletheia.app.ui.reader.tts

import org.junit.Assert.assertEquals
import org.junit.Test

class RussianSpeechTextSplitterTest {
    @Test
    fun keepsTerminalPunctuationAndDialogueDashInOneSpeechSentence() {
        listOf(
            "Стой. — сказал он.",
            "Стой! — сказал он.",
            "Стой? — спросил он.",
            "«Стой!» — сказал он.",
            "Стой? - спросил он."
        ).forEach { text ->
            assertEquals(listOf(text), RussianSpeechTextSplitter.splitSentences(text))
        }
    }

    @Test
    fun stillSeparatesOrdinarySentences() {
        assertEquals(
            listOf("Первое предложение.", "Второе предложение!", "Третье?"),
            RussianSpeechTextSplitter.splitSentences("Первое предложение. Второе предложение! Третье?")
        )
    }
}
