more test case

Author: bact

pythainlp/rank/__init__.py

@@ -18,9 +18,7 @@ def rank(words, stopword=False):
     if stopword:
         words = [word for word in words if word not in _STOPWORDS]
 
-    rankdata = Counter(words)
-
-    return rankdata
+    return Counter(words)
 
 
 if __name__ == "__main__":

tests/__init__.py

@@ -216,7 +216,7 @@ def test_romanization(self):
 
     def test_romanization_royin(self):
         engine = "royin"
-        self.assertIsNone(romanize("กก", engine=engine))
+        self.assertIsNotNone(romanize("กก", engine=engine))
         self.assertEqual(romanize("แมว", engine=engine), "maeo")
         self.assertEqual(romanize("เดือน", engine=engine), "duean")
         self.assertEqual(romanize("ดู", engine=engine), "du")
@@ -352,6 +352,7 @@ def test_deletetone(self):
 
     def test_is_thai(self):
         self.assertEqual(is_thai("ประเทศไทย"), {"thai": 100.0})
+        self.assertIsNotNone(is_thai("เผือก", check_all=True))
 
     def test_is_thaichar(self):
         self.assertEqual(is_thaichar("ก"), True)