Update pythainlp.corpus.find_synonyms

PyThaiNLP · Dec 12, 2023 · cf6647e · cf6647e
1 parent a5aa12c
commit cf6647e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pythainlp/corpus/common.py b/pythainlp/corpus/common.py
@@ -357,10 +357,10 @@ def find_synonyms(word: str) -> List[str]:
         # output: ['จรุก', 'วราห์', 'วราหะ', 'ศูกร', 'สุกร']
     """
     synonyms = thai_synonyms()  # get a dictionary of {word, synonym}
+    list_synonym = []
 
-    if word in synonyms["word"]:
-        # returns the position of the first occurrence of the word
-        idx = synonyms["word"].index(word)
-        return synonyms["synonym"][idx]
+    for idx, words in enumerate(synonyms["word"]):
+        if word in words:
+            list_synonym.extend(synonyms["synonym"][idx])
 
-    return []
+    return sorted(list(set(list_synonym)))
diff --git a/tests/test_corpus.py b/tests/test_corpus.py
@@ -209,6 +209,6 @@ def test_zip(self):
     def test_find_synonyms(self):
         self.assertEqual(
             find_synonyms("หมู"),
-            ['จรุก', 'วราห์', 'วราหะ', 'ศูกร', 'สุกร']
+            ['จรุก', 'วราหะ', 'วราห์', 'ศูกร', 'สุกร']
         )
         self.assertEqual(find_synonyms("1"), [])