class TestTextGenerator(unittest.TestCase):
def setUp(self):
self.generator = TextGenerator(capitalize = True)
def test_simple_generation(self):
t = "{1|2} {3|4}"
need = sorted(["1 3", "1 4", "2 3", "2 4"])
res = sorted(self.generator.generate(t))
self.assertEqual(need, res)
def test_complex_generation(self):
t = "{1|2} %kw% {3|4}"
need = sorted(["1 %kw% 3", "1 %kw% 4", "2 %kw% 3", "2 %kw% 4"])
res = sorted(self.generator.generate(t))
self.assertEqual(need, res)
def test_capitalizing(self):
text = "i want to born. hey, lala ley! what?damn, it! yes!!! da?! v"
cap_text = self.generator.smart_capitalize(text)
self.assertEqual(cap_text, "I want to born. Hey, lala ley! What?Damn, it! Yes!!! Da?! V")
class TestTextGenerator(unittest.TestCase):
def setUp(self):
self.generator = TextGenerator(capitalize=True)
def test_simple_generation(self):
t = "{1|2} {3|4}"
need = sorted(["1 3", "1 4", "2 3", "2 4"])
res = sorted(self.generator.generate(t))
self.assertEqual(need, res)
def test_complex_generation(self):
t = "{1|2} %kw% {3|4}"
need = sorted(["1 %kw% 3", "1 %kw% 4", "2 %kw% 3", "2 %kw% 4"])
res = sorted(self.generator.generate(t))
self.assertEqual(need, res)
def test_capitalizing(self):
text = "i want to born. hey, lala ley! what?damn, it! yes!!! da?! v"
cap_text = self.generator.smart_capitalize(text)
self.assertEqual(
cap_text,
"I want to born. Hey, lala ley! What?Damn, it! Yes!!! Da?! V")
def setUp(self):
self.generator = TextGenerator(capitalize=True)
def setUp(self):
self.generator = TextGenerator(capitalize = True)
