Fixed #1004

PyThaiNLP · Dec 25, 2024 · 6f1dc92 · 6f1dc92
1 parent 188904f
commit 6f1dc92
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 19 deletions.
diff --git a/pythainlp/util/syllable.py b/pythainlp/util/syllable.py
@@ -184,7 +184,7 @@ def syllable_length(syllable: str) -> str:
         # output: short
     """
     consonants = [i for i in syllable if i in list(thai_consonants)]
-    if len(consonants) < 3 and any((c in set(short)) for c in syllable):
+    if len(consonants) <= 3 and any((c in set(short)) for c in syllable):
         return "short"
     elif bool(re_short.search(syllable)):
         return "short"
@@ -259,15 +259,15 @@ def tone_detector(syllable: str) -> str:
             and tone_mark == "่"
         ):
             r = "l"
-        elif initial_consonant == "อ" and consonant_ending and s == "dead":
-            r = "l"
         elif (
             initial_consonant == "ห"
             and consonant_ending
             and s == "live"
             and tone_mark == "่"
         ):
             r = "l"
+        elif initial_consonant == "อ" and consonant_ending and s == "dead":
+            r = "l"
         elif (
             initial_consonant == "ห"
             and consonant_ending
@@ -279,6 +279,22 @@ def tone_detector(syllable: str) -> str:
             r = "l"
         elif initial_consonant == "ห" and consonant_ending and s == "live":
             r = "r"
+    elif initial_consonant_type == "high" and s == "live" and tone_mark == "่":
+        r = "l"
+    elif initial_consonant_type == "mid" and s == "live" and tone_mark == "่":
+        r = "l"
+    elif initial_consonant_type == "low" and tone_mark == "้":
+        r = "h"
+    elif initial_consonant_type == "mid" and tone_mark == "๋":
+        r = "r"
+    elif initial_consonant_type == "mid" and tone_mark == "๊":
+        r = "h"
+    elif initial_consonant_type == "low" and tone_mark == "่":
+        r = "f"
+    elif initial_consonant_type == "mid" and tone_mark == "้":
+        r = "f"
+    elif initial_consonant_type == "high" and tone_mark == "้":
+        r = "f"
     elif (
         initial_consonant_type == "low"
         and syllable_check_length == "short"
@@ -299,22 +315,6 @@ def tone_detector(syllable: str) -> str:
         and syllable_check == "open"
     ):
         r = "h"
-    elif initial_consonant_type == "high" and s == "live" and tone_mark == "่":
-        r = "l"
-    elif initial_consonant_type == "mid" and s == "live" and tone_mark == "่":
-        r = "l"
-    elif initial_consonant_type == "low" and tone_mark == "้":
-        r = "h"
-    elif initial_consonant_type == "mid" and tone_mark == "๋":
-        r = "r"
-    elif initial_consonant_type == "mid" and tone_mark == "๊":
-        r = "h"
-    elif initial_consonant_type == "low" and tone_mark == "่":
-        r = "f"
-    elif initial_consonant_type == "mid" and tone_mark == "้":
-        r = "f"
-    elif initial_consonant_type == "high" and tone_mark == "้":
-        r = "f"
     elif initial_consonant_type == "mid" and s == "dead":
         r = "l"
     elif initial_consonant_type == "high" and s == "dead":

diff --git a/tests/core/test_util.py b/tests/core/test_util.py
@@ -708,6 +708,8 @@ def test_tone_detector(self):
             ("f", "ไหม้"),
             ("f", "ต้น"),
             ("f", "ผู้"),
+            ("h", "ครับ"),
+            ("f", "ค่ะ"),
         ]
         for i, j in data:
             self.assertEqual(tone_detector(j), i)