more test cases

Author: bact

pythainlp/transliterate/__init__.py

@@ -3,9 +3,9 @@
 from pythainlp.tokenize import word_tokenize
 
 
-# ถอดเสียงภาษาไทยเป็นอักษรละติน
 def romanize(text: str, engine: str = "royin") -> str:
     """
+    ถอดเสียงภาษาไทยเป็นอักษรละติน
     :param str text: Thai text to be romanized
     :param str engine: 'royin' (default) or 'thai2rom'. 'royin' uses Thai Royal Institute standard. 'thai2rom' is deep learning Thai romanization (require keras).
     :return: English (more or less) text that spells out how the Thai text should read.
@@ -21,11 +21,9 @@ def romanize(text: str, engine: str = "royin") -> str:
     else:  # use default engine "royin"
         from .royin import romanize
 
-        try:
-            words = word_tokenize(text)
-            romanized_words = [romanize(word) for word in words]
-        except:
-            romanized_words = [romanize(text)]
+        words = word_tokenize(text)
+        romanized_words = [romanize(word) for word in words]
+
         return "".join(romanized_words)
 
 

pythainlp/transliterate/royin.py

@@ -176,5 +176,5 @@ def romanize(word: str) -> str:
         word2 = "".join(word2)
 
     word2 = _replace_consonants(word2, res)
-
+    
     return word2

tests/__init__.py

@@ -562,7 +562,10 @@ def test_normalize(self):
     # ### pythainlp.util.thai
 
     def test_countthai(self):
+        self.assertEqual(countthai(""), 0)
         self.assertEqual(countthai("ประเทศไทย"), 100.0)
+        self.assertEqual(countthai("(กกต.)", ".()"), 100.0)
+        self.assertEqual(countthai("(กกต.)", None), 50.0)
 
     def test_isthaichar(self):
         self.assertEqual(isthaichar("ก"), True)
@@ -574,6 +577,7 @@ def test_isthai(self):
         self.assertEqual(isthai("ไทย0"), False)
         self.assertEqual(isthai("ต.ค."), True)
         self.assertEqual(isthai("(ต.ค.)"), False)
+        self.assertEqual(isthai("ต.ค.", ignore_chars=None), False)
         self.assertEqual(isthai("(ต.ค.)", ignore_chars=".()"), True)
 
     def test_is_thaicheck(self):