61 lines
2.6 KiB
Python
61 lines
2.6 KiB
Python
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."}
|