Add more cases

Author: bact

pythainlp/khavee/core.py

@@ -219,7 +219,7 @@ def check_marttra(self, word: str) -> str:
 
             kv = KhaveeVerifier()
 
-            print(kv.check_marttra('สาว'))
+            print(kv.check_marttra("สาว"))
             # output: 'เกอว'
         """
         if word[-1] == "ร" and word[-2] in ["ต", "ท"]:
@@ -297,10 +297,10 @@ def is_sumpus(self, word1: str, word2: str) -> bool:
 
             kv = KhaveeVerifier()
 
-            print(kv.is_sumpus('สรร', 'อัน'))
+            print(kv.is_sumpus("สรร", "อัน"))
             # output: True
 
-            print(kv.is_sumpus('สรร', 'แมว'))
+            print(kv.is_sumpus("สรร", "แมว"))
             # output: False
         """
         marttra1 = self.check_marttra(word1)
@@ -609,9 +609,13 @@ def check_aek_too(
             kv = KhaveeVerifier()
 
             # การเช็คคำเอกโท
-            print(kv.check_aek_too('เอง'), kv.check_aek_too('เอ่ง'), kv.check_aek_too('เอ้ง'))
+            print(
+                kv.check_aek_too("เอง"),
+                kv.check_aek_too("เอ่ง"),
+                kv.check_aek_too("เอ้ง"),
+            )
             # -> False, aek, too
-            print(kv.check_aek_too(['เอง', 'เอ่ง', 'เอ้ง'])) # ใช้ List ได้เหมือนกัน
+            print(kv.check_aek_too(["เอง", "เอ่ง", "เอ้ง"]))  # ใช้ List ได้เหมือนกัน
             # -> [False, 'aek', 'too']
 
 

tests/test_khavee.py

@@ -15,21 +15,71 @@ def test_check_sara(self):
 
     def test_check_marttra(self):
         self.assertEqual(kv.check_marttra("ปลิง"), "กง")
+        self.assertEqual(kv.check_marttra("ยูง"), "กง")
+        self.assertEqual(kv.check_marttra("กล่อง"), "กง")
+        self.assertEqual(kv.check_marttra("สอง"), "กง")
+        self.assertEqual(kv.check_marttra("เอ็ง"), "กง")
+        self.assertEqual(kv.check_marttra("งง"), "กง")
+
         self.assertEqual(kv.check_marttra("ลม"), "กม")
+        self.assertEqual(kv.check_marttra("เฉลิม"), "กม")
+        self.assertEqual(kv.check_marttra("เข็ม"), "กม")
+        self.assertEqual(kv.check_marttra("จาม"), "กม")
+        self.assertEqual(kv.check_marttra("ยิ้ม"), "กม")
+        self.assertEqual(kv.check_marttra("เกม"), "กม")
+        # self.assertEqual(kv.check_marttra("รมย์"), "กม")
+
+        self.assertEqual(kv.check_marttra("สวย"), "เกย")
         self.assertEqual(kv.check_marttra("โปรย"), "เกย")
+        self.assertEqual(kv.check_marttra("เนย"), "เกย")
+        self.assertEqual(kv.check_marttra("คอย"), "เกย")
+        self.assertEqual(kv.check_marttra("ง่าย"), "เกย")
+        # self.assertEqual(kv.check_marttra("ทัย"), "เกย")
+        # self.assertEqual(kv.check_marttra("ไทย"), "เกย")
+        # self.assertEqual(kv.check_marttra("ไกล"), "เกย")
+        # self.assertEqual(kv.check_marttra("ใกล้"), "เกย")
+
         self.assertEqual(kv.check_marttra("สาว"), "เกอว")
+        self.assertEqual(kv.check_marttra("นิ้ว"), "เกอว")
+        self.assertEqual(kv.check_marttra("แมว"), "เกอว")
+        self.assertEqual(kv.check_marttra("ดาว"), "เกอว")
+        self.assertEqual(kv.check_marttra("แก้ว"), "เกอว")
+
         self.assertEqual(kv.check_marttra("บก"), "กก")
         self.assertEqual(kv.check_marttra("โรค"), "กก")
+        self.assertEqual(kv.check_marttra("ลาก"), "กก")
+        self.assertEqual(kv.check_marttra("นัข"), "กก")
         # self.assertEqual(kv.check_marttra("จักร"), "กก")
+
         self.assertEqual(kv.check_marttra("จด"), "กด")
         self.assertEqual(kv.check_marttra("ตรวจ"), "กด")
+        self.assertEqual(kv.check_marttra("เสริฐ"), "กด")
+        self.assertEqual(kv.check_marttra("บุตร"), "กด")
+        self.assertEqual(kv.check_marttra("ตรุษ"), "กด")
+        # self.assertEqual(kv.check_marttra("โยชน์"), "กด")
+
+        self.assertEqual(kv.check_marttra("ร้าน"), "กน")
+        self.assertEqual(kv.check_marttra("ขนุน"), "กน")
         self.assertEqual(kv.check_marttra("คน"), "กน")
         self.assertEqual(kv.check_marttra("ทมิฬ"), "กน")
+        self.assertEqual(kv.check_marttra("ซีน"), "กน")
         self.assertEqual(kv.check_marttra("บรร"), "กน")
         self.assertEqual(kv.check_marttra("กร"), "กน")
+        self.assertEqual(kv.check_marttra("เณร"), "กน")
+        # self.assertEqual(kv.check_marttra("ยนต์"), "กน")
+        # self.assertEqual(kv.check_marttra("กรรณ"), "กน")
+
         self.assertEqual(kv.check_marttra("ชอบ"), "กบ")
         self.assertEqual(kv.check_marttra("ภาพ"), "กบ")
+        self.assertEqual(kv.check_marttra("เทพ"), "กบ")
+        self.assertEqual(kv.check_marttra("รูป"), "กบ")
+        self.assertEqual(kv.check_marttra("เวฟ"), "กบ")
+        self.assertEqual(kv.check_marttra("โลพ"), "กบ")
+
         self.assertEqual(kv.check_marttra("ปลา"), "กา")
+        self.assertEqual(kv.check_marttra("งู"), "กา")
+        self.assertEqual(kv.check_marttra("หมู"), "กา")
+        self.assertEqual(kv.check_marttra("มือ"), "กา")
 
     def test_is_sumpus(self):
         self.assertTrue(kv.is_sumpus("สรร", "อัน"))