Update test suites

Author: bact

pythainlp/spell/__init__.py

@@ -11,12 +11,12 @@
     "correct",
     "correct_sent",
     "spell",
-    "SENT_TOKS",
+    "spell_sent",
 ]
 
 from pythainlp.spell.pn import NorvigSpellChecker
 
 DEFAULT_SPELL_CHECKER = NorvigSpellChecker()
 
 # these imports are placed here to avoid circular imports
-from pythainlp.spell.core import correct, correct_sent, spell, SENT_TOKS
+from pythainlp.spell.core import correct, correct_sent, spell, spell_sent

pythainlp/spell/core.py

@@ -129,7 +129,7 @@ def correct(word: str, engine: str = "pn") -> str:
     return text_correct
 
 
-def SENT_TOKS(list_words: List[str], engine: str = "pn") -> List[List[str]]:
+def spell_sent(list_words: List[str], engine: str = "pn") -> List[List[str]]:
     """
     Provides a list of possible correct spellings of sentence
 
@@ -188,4 +188,4 @@ def correct_sent(list_words: List[str], engine: str = "pn") -> List[str]:
         correct_sent(["เด็","อินอร์เน็ต","แรง"],engine='symspellpy')
         # output: ['เด็ก', 'อินเทอร์เน็ต', 'แรง']
     """
-    return SENT_TOKS(list_words, engine=engine)[0]
+    return spell_sent(list_words, engine=engine)[0]

tests/__init__.py

@@ -8,15 +8,17 @@
 """
 
 import sys
-from unittest import TestLoader, TestSuite, TextTestRunner
-
+from unittest import TestLoader, TestSuite
 
 test_packages = [
+    "tests.test_cli.TestMainPackage",
+    "tests.test_soundex.TestSoundexPackage",
     "tests.test_spell.TestSpellPackage",
     "tests.test_tokenize.TestTokenizePackage",
     "tests.test_util.TestUtilPackage",
 ]
 
+
 def load_tests(loader: TestLoader, tests, pattern: str) -> TestSuite:
     """A function to load tests."""
     suite = TestSuite()

tests/test_tokenize.py

@@ -214,7 +214,7 @@ def test_Tokenizer(self):
         _tokenizer = Tokenizer()
         self.assertEqual(_tokenizer.word_tokenize("ก"), ["ก"])
         with self.assertRaises(NotImplementedError):
-            Tokenizer(engine="catcut")
+            Tokenizer(engine="catcut888")
 
     def test_sent_tokenize(self):
         self.assertEqual(sent_tokenize(None), [])