Spaces:
Paused
Paused
File size: 1,406 Bytes
45ee559 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import unittest
from TTS.tts.utils.text.punctuation import _DEF_PUNCS, Punctuation
class PunctuationTest(unittest.TestCase):
def setUp(self):
self.punctuation = Punctuation()
self.test_texts = [
("This, is my text ... to be striped !! from text?", "This is my text to be striped from text"),
("This, is my text ... to be striped !! from text", "This is my text to be striped from text"),
("This, is my text ... to be striped from text?", "This is my text to be striped from text"),
("This, is my text to be striped from text", "This is my text to be striped from text"),
]
def test_get_set_puncs(self):
self.punctuation.puncs = "-="
self.assertEqual(self.punctuation.puncs, "-=")
self.punctuation.puncs = _DEF_PUNCS
self.assertEqual(self.punctuation.puncs, _DEF_PUNCS)
def test_strip_punc(self):
for text, gt in self.test_texts:
text_striped = self.punctuation.strip(text)
self.assertEqual(text_striped, gt)
def test_strip_restore(self):
for text, gt in self.test_texts:
text_striped, puncs_map = self.punctuation.strip_to_restore(text)
text_restored = self.punctuation.restore(text_striped, puncs_map)
self.assertEqual(" ".join(text_striped), gt)
self.assertEqual(text_restored[0], text)
|