Move extras to testx

Author: bact

pythainlp/spell/__init__.py

@@ -11,12 +11,12 @@
     "correct",
     "correct_sent",
     "spell",
-    "spell_sent",
+    "SENT_TOKS",
 ]
 
 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, spell_sent
+from pythainlp.spell.core import correct, correct_sent, spell, SENT_TOKS

pythainlp/spell/core.py

@@ -129,7 +129,7 @@ def correct(word: str, engine: str = "pn") -> str:
     return text_correct
 
 
-def spell_sent(list_words: List[str], engine: str = "pn") -> List[List[str]]:
+def SENT_TOKS(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 spell_sent(list_words, engine=engine)[0]
+    return SENT_TOKS(list_words, engine=engine)[0]

tests/__init__.py

@@ -11,10 +11,16 @@
 from unittest import TestLoader, TestSuite, TextTestRunner
 
 
-def load_tests(loader: TestLoader, tests, pattern) -> TestSuite:
+test_packages = [
+    "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 = loader.loadTestsFromName("tests.test_util.TestUtilPackage")
-    suite.addTests(tests)
-    tests = loader.loadTestsFromName("tests.test_tokenize.TestTokenizePackage")
-    suite.addTests(tests)
+    for test_package in test_packages:
+        tests = loader.loadTestsFromName(test_package)
+        suite.addTests(tests)
     return suite

tests/test_spell.py

@@ -10,9 +10,9 @@
     correct_sent,
     spell,
     spell_sent,
-    symspellpy,
 )
 
+SENT_TOKS = ["เด็", "อินอร์เน็ต", "แรง"]
 
 class TestSpellPackage(unittest.TestCase):
     def test_spell(self):
@@ -27,30 +27,6 @@ def test_spell(self):
         self.assertIsInstance(result, list)
         self.assertGreater(len(result), 0)
 
-        result = spell("เน้ร", engine="phunspell")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
-        result = spell("เกสมร์", engine="phunspell")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
-        result = spell("เน้ร", engine="symspellpy")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
-        result = spell("เกสมร์", engine="symspellpy")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
-        result = spell("เน้ร", engine="tltk")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
-        result = spell("เดก", engine="tltk")
-        self.assertIsInstance(result, list)
-        self.assertGreater(len(result), 0)
-
     def test_word_correct(self):
         self.assertEqual(correct(None), "")
         self.assertEqual(correct(""), "")
@@ -63,18 +39,6 @@ def test_word_correct(self):
         self.assertIsInstance(result, str)
         self.assertNotEqual(result, "")
 
-        result = correct("ทดสอง", engine="phunspell")
-        self.assertIsInstance(result, str)
-        self.assertNotEqual(result, "")
-
-        result = correct("ทดสอง", engine="symspellpy")
-        self.assertIsInstance(result, str)
-        self.assertNotEqual(result, "")
-
-        result = correct("ทดสอง", engine="wanchanberta_thai_grammarly")
-        self.assertIsInstance(result, str)
-        self.assertNotEqual(result, "")
-
     def test_norvig_spell_checker(self):
         checker = NorvigSpellChecker(dict_filter=None)
         self.assertTrue(len(checker.dictionary()) > 0)
@@ -124,21 +88,10 @@ def test_norvig_spell_checker(self):
             checker = NorvigSpellChecker(custom_dict=user_dict)
 
     def test_spell_sent(self):
-        self.spell_sent = ["เด็", "อินอร์เน็ต", "แรง"]
-        self.assertIsNotNone(spell_sent(self.spell_sent))
-        self.assertIsNotNone(spell_sent(self.spell_sent, engine="pn"))
-        self.assertIsNotNone(spell_sent(self.spell_sent, engine="phunspell"))
-        self.assertIsNotNone(spell_sent(self.spell_sent, engine="symspellpy"))
+        self.assertIsNotNone(spell_sent(SENT_TOKS))
+        self.assertIsNotNone(spell_sent(SENT_TOKS, engine="pn"))
 
     def test_correct_sent(self):
-        self.spell_sent = ["เด็", "อินอร์เน็ต", "แรง"]
-        self.assertIsNotNone(correct_sent(self.spell_sent))
-        self.assertIsNotNone(correct_sent(self.spell_sent, engine="pn"))
-        self.assertIsNotNone(correct_sent(self.spell_sent, engine="phunspell"))
-        self.assertIsNotNone(
-            correct_sent(self.spell_sent, engine="symspellpy")
-        )
-        self.assertIsNotNone(
-            correct_sent(self.spell_sent, engine="wanchanberta_thai_grammarly")
-        )
-        self.assertIsNotNone(symspellpy.correct_sent(self.spell_sent))
+        self.assertIsNotNone(correct_sent(SENT_TOKS))
+        self.assertIsNotNone(correct_sent(SENT_TOKS, engine="pn"))
+

tests/test_augment.py renamed to tests/testx_augment.py

@@ -0,0 +1,67 @@
+# -*- coding: utf-8 -*-
+# SPDX-FileCopyrightText: 2016-2024 PyThaiNLP Project
+# SPDX-License-Identifier: Apache-2.0
+
+import unittest
+
+from pythainlp.spell import (
+    correct,
+    correct_sent,
+    spell,
+    spell_sent,
+    symspellpy,
+)
+
+SENT_TOKS = ["เด็", "อินอร์เน็ต", "แรง"]
+
+
+class TestSpellPackage(unittest.TestCase):
+    def test_spell(self):
+        result = spell("เน้ร", engine="phunspell")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+        result = spell("เกสมร์", engine="phunspell")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+        result = spell("เน้ร", engine="symspellpy")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+        result = spell("เกสมร์", engine="symspellpy")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+        result = spell("เน้ร", engine="tltk")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+        result = spell("เดก", engine="tltk")
+        self.assertIsInstance(result, list)
+        self.assertGreater(len(result), 0)
+
+    def test_word_correct(self):
+        result = correct("ทดสอง", engine="phunspell")
+        self.assertIsInstance(result, str)
+        self.assertNotEqual(result, "")
+
+        result = correct("ทดสอง", engine="symspellpy")
+        self.assertIsInstance(result, str)
+        self.assertNotEqual(result, "")
+
+        result = correct("ทดสอง", engine="wanchanberta_thai_grammarly")
+        self.assertIsInstance(result, str)
+        self.assertNotEqual(result, "")
+
+    def test_spell_sent(self):
+        self.assertIsNotNone(spell_sent(SENT_TOKS, engine="phunspell"))
+        self.assertIsNotNone(spell_sent(SENT_TOKS, engine="symspellpy"))
+
+    def test_correct_sent(self):
+        self.assertIsNotNone(correct_sent(SENT_TOKS, engine="phunspell"))
+        self.assertIsNotNone(correct_sent(SENT_TOKS, engine="symspellpy"))
+        self.assertIsNotNone(
+            correct_sent(SENT_TOKS, engine="wanchanberta_thai_grammarly")
+        )
+        self.assertIsNotNone(symspellpy.correct_sent(SENT_TOKS))