tts-markup-utility/test_main.py

from unittest import TestCase

from main import SimpleMarkupParser


class Test(TestCase):
    def test_simple_markup_parser_0(self):
        # Test case with sections
        markup_text = "[section 1] [voice alloy] Hello, this is section 1. [end_section] [voice nova] Now we're outside the section. [insert_section 1]"
        parser = SimpleMarkupParser(markup_text)
        parser.parse()
        parsed_output = parser.get_output()

        assert len(parsed_output) == 7, "Expected 7 tokens, got %d" % len(parsed_output)
        assert parsed_output[0] == {"type": "section_start", "section_id": 1}
        assert parsed_output[1] == {"type": "voice", "voice": "alloy"}
        assert parsed_output[2] == {"type": "text", "text": "Hello, this is section 1."}
        assert parsed_output[3] == {"type": "section_end"}
        assert parsed_output[4] == {"type": "voice", "voice": "nova"}
        assert parsed_output[5] == {
            "type": "text",
            "text": "Now we're outside the section.",
        }
        assert parsed_output[6] == {"type": "insert_section", "section_id": 1}

    def test_simple_markup_parser_1(self):
        # Test case with silence
        markup_text = (
            "[voice nova] Let's have a moment of silence. [silence 3s] And we're back!"
        )
        parser = SimpleMarkupParser(markup_text)
        parser.parse()
        parsed_output = parser.get_output()

        assert len(parsed_output) == 4
        assert parsed_output[0] == {"type": "voice", "voice": "nova"}
        assert parsed_output[1] == {
            "type": "text",
            "text": "Let's have a moment of silence.",
        }
        assert parsed_output[2] == {"type": "silence", "duration": 3000}
        assert parsed_output[3] == {"type": "text", "text": "And we're back!"}

    def test_simple_markup_parser_2(self):
        # Test case with unknown markup
        markup_text = "[voice fable] Hello! [unknown_markup] This is an unknown markup. [voice nova] Back to a known voice."
        parser = SimpleMarkupParser(markup_text)
        parser.parse()
        parsed_output = parser.get_output()

        assert len(parsed_output) == 6
        assert parsed_output[0] == {"type": "voice", "voice": "fable"}
        assert parsed_output[1] == {"type": "text", "text": "Hello!"}
        assert parsed_output[2] == {"type": "none", "text": "[unknown_markup]"}
        assert parsed_output[3] == {
            "type": "text",
            "text": "This is an unknown markup.",
        }
        assert parsed_output[4] == {"type": "voice", "voice": "nova"}
        assert parsed_output[5] == {"type": "text", "text": "Back to a known voice."}