Close some test to save space

PyThaiNLP · Aug 14, 2023 · e93c0be · e93c0be
1 parent 9c9f881
commit e93c0be
Show file tree

Hide file tree

Showing 6 changed files with 69 additions and 81 deletions.
diff --git a/tests/test_augment.py b/tests/test_augment.py
@@ -3,12 +3,12 @@
 import unittest
 from pythainlp.augment import WordNetAug
 from pythainlp.augment.wordnet import postype2wordnet
-from pythainlp.augment.lm import Thai2transformersAug
+# from pythainlp.augment.lm import Thai2transformersAug
 from pythainlp.augment.word2vec.bpemb_wv import BPEmbAug
 from pythainlp.augment.word2vec import (
-    Thai2fitAug,
     LTW2VAug
 )
+# Thai2fitAug,
 import nltk
 
 
@@ -26,10 +26,10 @@ def test_WordNetAug(self):
         self.assertIsNone(postype2wordnet('n', 'abc'))
         self.assertIsNotNone(postype2wordnet('NOUN', 'orchid'))
 
-    def test_Thai2fitAug(self):
-        _aug = Thai2fitAug()
-        self.assertIsNotNone(_aug.tokenizer(self.text))
-        self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
+    # def test_Thai2fitAug(self):
+    #     _aug = Thai2fitAug()
+    #     self.assertIsNotNone(_aug.tokenizer(self.text))
+    #     self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
 
     def test_BPEmbAug(self):
         _aug = BPEmbAug()
@@ -41,6 +41,6 @@ def test_LTW2VAug(self):
         self.assertIsNotNone(_aug.tokenizer(self.text))
         self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
 
-    def test_Thai2transformersAug(self):
-        _aug = Thai2transformersAug()
-        self.assertIsNotNone(_aug.augment(self.text2, num_replace_tokens=1))
+    # def test_Thai2transformersAug(self):
+    #     _aug = Thai2transformersAug()
+    #     self.assertIsNotNone(_aug.augment(self.text2, num_replace_tokens=1))
diff --git a/tests/test_generate.py b/tests/test_generate.py
@@ -40,7 +40,7 @@ def test_trigram(self):
         self.assertIsNotNone(_trigram.gen_sentence())
         self.assertIsNotNone(_trigram.gen_sentence(duplicate=True))
 
-    def test_thai2fit(self):
-        self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว"))
-        self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว", output_str=False))
-        self.assertIsNotNone(gen_sentence())
+    # def test_thai2fit(self):
+    #     self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว"))
+    #     self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว", output_str=False))
+    #     self.assertIsNotNone(gen_sentence())
diff --git a/tests/test_parse.py b/tests/test_parse.py
@@ -8,9 +8,9 @@ class TestParsePackage(unittest.TestCase):
     def test_dependency_parsing(self):
         self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="esupar"))
         self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="esupar", tag="list"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud", tag="list"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai", tag="list"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith"))
-        self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith", tag="list"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud", tag="list"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai", tag="list"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith"))
+        # self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith", tag="list"))
diff --git a/tests/test_summarize.py b/tests/test_summarize.py
@@ -20,9 +20,9 @@ def test_summarize(self):
             summarize(text=text, n=1),
             ["อาหารจะต้องไม่มีพิษและไม่เกิดโทษต่อร่างกาย"],
         )
-        self.assertIsNotNone(summarize(text, engine="mt5-small"))
-        self.assertIsNotNone(summarize([]))
-        self.assertIsNotNone(summarize(text, 1, engine="mt5-small"))
+        # self.assertIsNotNone(summarize(text, engine="mt5-small"))
+        # self.assertIsNotNone(summarize([]))
+        # self.assertIsNotNone(summarize(text, 1, engine="mt5-small"))
         self.assertIsNotNone(summarize(text, 1, engine="mt5-cpe-kmutt-thai-sentence-sum"))
         self.assertIsNotNone(summarize(text, 1, engine="XX"))
         with self.assertRaises(ValueError):

diff --git a/tests/test_tag.py b/tests/test_tag.py
@@ -346,10 +346,10 @@ def test_NER_class(self):
         self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
         self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
         self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
-        ner = NER(engine="wangchanberta")
-        self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
-        self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
-        self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
+        # ner = NER(engine="wangchanberta")
+        # self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
+        # self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
+        # self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
         ner = NER(engine="thainer-v2")
         self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
         self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))

diff --git a/tests/test_translate.py b/tests/test_translate.py
@@ -17,7 +17,7 @@
 
 class TestTranslatePackage(unittest.TestCase):
     def test_translate(self):
-        remove("scb_1m_th-en_spm")
+        # remove("scb_1m_th-en_spm")
         self.assertIsNone(download_model_all())
         self.th_en_translator = ThEnTranslator()
         self.assertIsNotNone(
@@ -31,59 +31,47 @@ def test_translate(self):
                 "the cat eats fish.",
             )
         )
-        self.th_zh_translator = ThZhTranslator()
-        self.assertIsNotNone(
-            self.th_zh_translator.translate(
-                "ผมรักคุณ",
-            )
-        )
-        self.zh_th_translator = ZhThTranslator()
-        self.assertIsNotNone(
-            self.zh_th_translator.translate(
-                "我爱你",
-            )
-        )
-        self.th_en_translator = Translate('th', 'en')
-        self.assertIsNotNone(
-            self.th_en_translator.translate(
-                "แมวกินปลา",
-            )
-        )
-        self.en_th_translator = Translate('en', 'th')
-        self.assertIsNotNone(
-            self.en_th_translator.translate(
-                "the cat eats fish.",
-            )
-        )
-        self.th_zh_translator = Translate('th', 'zh')
-        self.assertIsNotNone(
-            self.th_zh_translator.translate(
-                "ผมรักคุณ",
-            )
-        )
-        self.zh_th_translator = Translate('zh', 'th')
-        self.assertIsNotNone(
-            self.zh_th_translator.translate(
-                "我爱你",
-            )
-        )
-        self.th_fr_translator = Translate('th', 'fr')
-        self.assertIsNotNone(
-            self.th_fr_translator.translate(
-                "ทดสอบระบบ",
-            )
-        )
-        self.th_fr_translator = Translate('th', 'fr', engine="small100")
-        self.assertIsNotNone(
-            self.th_fr_translator.translate(
-                "ทดสอบระบบ",
-            )
-        )
-        self.th_ja_translator = Translate('th', 'ja', engine="small100")
-        self.assertIsNotNone(
-            self.th_fr_translator.translate(
-                "ทดสอบระบบ",
-            )
-        )
+        # self.th_zh_translator = ThZhTranslator()
+        # self.assertIsNotNone(
+        #     self.th_zh_translator.translate(
+        #         "ผมรักคุณ",
+        #     )
+        # )
+        # self.zh_th_translator = ZhThTranslator()
+        # self.assertIsNotNone(
+        #     self.zh_th_translator.translate(
+        #         "我爱你",
+        #     )
+        # )
+        # self.th_zh_translator = Translate('th', 'zh')
+        # self.assertIsNotNone(
+        #     self.th_zh_translator.translate(
+        #         "ผมรักคุณ",
+        #     )
+        # )
+        # self.zh_th_translator = Translate('zh', 'th')
+        # self.assertIsNotNone(
+        #     self.zh_th_translator.translate(
+        #         "我爱你",
+        #     )
+        # )
+        # self.th_fr_translator = Translate('th', 'fr')
+        # self.assertIsNotNone(
+        #     self.th_fr_translator.translate(
+        #         "ทดสอบระบบ",
+        #     )
+        # )
+        # self.th_fr_translator = Translate('th', 'fr', engine="small100")
+        # self.assertIsNotNone(
+        #     self.th_fr_translator.translate(
+        #         "ทดสอบระบบ",
+        #     )
+        # )
+        # self.th_ja_translator = Translate('th', 'ja', engine="small100")
+        # self.assertIsNotNone(
+        #     self.th_fr_translator.translate(
+        #         "ทดสอบระบบ",
+        #     )
+        # )
         with self.assertRaises(ValueError):
             self.th_cat_translator = Translate('th', 'cat', engine="fkfj")