|
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() |
|
|