| import unittest | |
| from text.numbers_ca import normalize_numbers_ca | |
| class NumbersCa(unittest.TestCase): | |
| def test_cardinals(self): | |
| """ | |
| Converteix cardinals simples en una frase | |
| """ | |
| self.assertEqual(normalize_numbers_ca("Va nèixer el 23 de desembre de 1988"), "Va nèixer el vint-i-tres de desembre de mil nou-cents vuitanta-vuit") | |
| self.assertEqual(normalize_numbers_ca("tinc 3 preguntes"), "tinc tres preguntes") | |
| def test_separador_milers(self): | |
| """ | |
| Ignora separadors de milers | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1.000"), "mil") | |
| self.assertEqual(normalize_numbers_ca("323.400"), "tres-cents vint-i-tres mil quatre-cents") | |
| self.assertEqual(normalize_numbers_ca("900.323.400"), "nou-cents milions tres-cents vint-i-tres mil quatre-cents") | |
| def test_decimals(self): | |
| """ | |
| Converteix decimals | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1,33"), "u coma trenta-tres") | |
| self.assertEqual(normalize_numbers_ca("75,5"), "setanta-cinc coma cinc") | |
| self.assertEqual(normalize_numbers_ca("75,555"), "setanta-cinc coma cinc-cents cinquanta-cinc") | |
| self.assertEqual(normalize_numbers_ca("999.999.999,99"), "nou-cents noranta-nou milions nou-cents noranta-nou mil nou-cents noranta-nou coma noranta-nou") | |
| self.assertEqual(normalize_numbers_ca("1,12345678900"), "u coma dotze trenta-quatre cinquanta-sis set-cents vuitanta-nou") | |
| def test_decimals_2(self): | |
| """ | |
| Ignora comes que no pertànyen a un número decimal | |
| """ | |
| self.assertEqual(normalize_numbers_ca("Va comprar pa, vi i llonganisses"), "Va comprar pa, vi i llonganisses") | |
| self.assertEqual(normalize_numbers_ca("El número guanyador és 1, 23, 55, 34"), "El número guanyador és u, vint-i-tres, cinquanta-cinc, trenta-quatre") | |
| def test_ordinals_ms(self): | |
| """ | |
| Converteix ordinals masculins singulars | |
| """ | |
| self.assertEqual(normalize_numbers_ca("Va arribar 4t de 5"), "Va arribar quart de cinc") | |
| self.assertEqual(normalize_numbers_ca("el 1r va ser ell"), "el primer va ser ell") | |
| self.assertEqual(normalize_numbers_ca("el 3er, no va aguantar"), "el tercer, no va aguantar") | |
| self.assertEqual(normalize_numbers_ca("2n"), "segon") | |
| self.assertEqual(normalize_numbers_ca("2on"), "segon") | |
| self.assertEqual(normalize_numbers_ca("4t"), "quart") | |
| self.assertEqual(normalize_numbers_ca("4rt"), "quart") | |
| self.assertEqual(normalize_numbers_ca("5è: remogueu la barreja"), "cinquè: remogueu la barreja") | |
| self.assertEqual(normalize_numbers_ca("6e"), "sisè") | |
| self.assertEqual(normalize_numbers_ca("6e"), "sisè") | |
| self.assertEqual(normalize_numbers_ca("21nè"), "vint-i-unè") | |
| self.assertEqual(normalize_numbers_ca("un 81ne de Palamós"), "un vuitanta-unè de Palamós") | |
| def test_ordinals_fs(self): | |
| """ | |
| Converteix ordinals femenins singulars | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1a"), "primera") | |
| self.assertEqual(normalize_numbers_ca("3ra"), "tercera") | |
| self.assertEqual(normalize_numbers_ca("2a"), "segona") | |
| self.assertEqual(normalize_numbers_ca("2na"), "segona") | |
| self.assertEqual(normalize_numbers_ca("4a."), "quarta.") | |
| self.assertEqual(normalize_numbers_ca("pugi a la 4ta, després giri a l'esquerra"), "pugi a la quarta, després giri a l'esquerra") | |
| self.assertEqual(normalize_numbers_ca("va quedar 5a en la classificació"), "va quedar cinquena en la classificació") | |
| self.assertEqual(normalize_numbers_ca("la 5na vegada"), "la cinquena vegada") | |
| def test_ordinals_mp(self): | |
| """ | |
| Converteix ordinals masculins plurals | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1rs"), "primers") | |
| self.assertEqual(normalize_numbers_ca("van arribar 2ns"), "van arribar segons") | |
| def test_ordinals_fp(self): | |
| """ | |
| Converteix ordinals femenins plurals | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1es"), "primeres") | |
| def test_fraccions_s(self): | |
| """ | |
| Converteix fraccions singulars | |
| """ | |
| self.assertEqual(normalize_numbers_ca("1/2 got de vi"), "mig got de vi") | |
| self.assertEqual(normalize_numbers_ca("1/3 de farina"), "un terç de farina") | |
| self.assertEqual(normalize_numbers_ca("1/8"), "un vuitè") | |
| def test_fraccions_p(self): | |
| """ | |
| Converteix fraccions plurals | |
| """ | |
| self.assertEqual(normalize_numbers_ca("4/2 gots de vi"), "quatre migs gots de vi") | |
| self.assertEqual(normalize_numbers_ca("2/3 de farina"), "dos terços de farina") | |
| self.assertEqual(normalize_numbers_ca("3/8"), "tres vuitens") | |
| def test_hores(self): | |
| """ | |
| Converteix hores de manera simplificada | |
| """ | |
| self.assertEqual(normalize_numbers_ca("a les 11:45"), "a les onze i quaranta-cinc") | |
| self.assertEqual(normalize_numbers_ca("a partir de les 23:12"), "a partir de les vint-i-tres i dotze") | |
| if __name__ == '__main__': | |
| unittest.main() | |