Fix gender_stem initialisation and add unittest for spanish

issue7
rohithkodali · Aug 27, 2013 · 29f31eb · 29f31eb
1 parent 8cb479d
commit 29f31eb
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 11 deletions.
diff --git a/numword/numword_es.py b/numword/numword_es.py
@@ -17,8 +17,8 @@ class NumWordES(NumWordEU):
     '''
 
     def __init__(self):
+        self.gender_stem = None
         super(NumWordES, self).__init__()
-        self.gender_stem = ''
 
     #TODO Is this sufficient??
     def _set_high_numwords(self, high):
@@ -52,16 +52,16 @@ def _setup(self):
             (30, u"treinta"),
             ]
         self.low_numwords = [
-            u"vientinueve",
-            u"vientiocho",
-            u"vientisiete",
-            u"vientiséis",
-            u"vienticinco",
-            u"vienticuatro",
-            u"vientitrés",
-            u"vientidós",
-            u"vientiuno",
-            u"viente",
+            u"veintinueve",
+            u"veintiocho",
+            u"veintisiete",
+            u"veintiséis",
+            u"veinticinco",
+            u"veinticuatro",
+            u"veintitrés",
+            u"veintidós",
+            u"veintiuno",
+            u"veinte",
             u"diecinueve",
             u"dieciocho",
             u"diecisiete",

diff --git a/numword/tests.py b/numword/tests.py
@@ -226,5 +226,33 @@ def test_ordinal(self):
         for number, word in tests:
             self.assertEqual(ordinal(number), word)
 
+
+class TestNumWordES(TestCase):
+
+    def test_cardinal(self):
+        from numword.numword_es import cardinal
+        self.assertEqual(cardinal(0), u"cero")
+        self.assertEqual(cardinal(11.96), u"once punto noventa y seis")
+        self.assertEqual(cardinal(100), u"cien")
+        self.assertEqual(cardinal(100.0), u"cien")
+        self.assertEqual(cardinal(121.01), u"ciento veintiuno punto uno")
+        self.assertEqual(cardinal(3121.45),
+            u"tres mil ciento veintiuno punto cuarenta y cinco")
+
+    def test_cardinal_not_a_number(self):
+        from numword.numword_es import cardinal
+        error = u"type\(Ximinez\) not in \[long, int, float\]"
+        with self.assertRaisesRegexp(TypeError, error):
+            cardinal('Ximinez')
+
+    def test_cardinal_number_too_big(self):
+        from numword.numword_es import cardinal
+        from numword.numword_es import NumWordES
+        max_val = NumWordES().maxval
+        number = max_val + 1
+        error = u"abs\(%s\) must be less than %s" % (number, max_val)
+        with self.assertRaisesRegexp(OverflowError, error):
+            cardinal(number)
+
 if __name__ == '__main__':
     main()