Fixed KhaveeVerifier bug

PyThaiNLP · Apr 14, 2023 · 356e47d · 356e47d
1 parent 454ead4
commit 356e47d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pythainlp/khavee/core.py b/pythainlp/khavee/core.py
@@ -14,6 +14,7 @@
 # limitations under the License.
 from typing import List, Union
 from pythainlp.tokenize import subword_tokenize
+from pythainlp.util import sound_syllable
 
 
 class KhaveeVerifier:

diff --git a/tests/test_khavee.py b/tests/test_khavee.py
@@ -12,9 +12,9 @@ def test_check_sara(self):
     def test_check_marttra(self):
         self.assertEqual(kv.check_marttra('สาว'), 'เกอว')
 
-    def test_check_sumpus(self):
-        self.assertTrue(kv.check_sumpus('สรร','อัน'))
-        self.assertFalse(kv.check_sumpus('สรร','แมว'))
+    def test_is_sumpus(self):
+        self.assertTrue(kv.is_sumpus('สรร','อัน'))
+        self.assertFalse(kv.is_sumpus('สรร','แมว'))
 
     def test_check_klon(self):
         self.assertEqual(