Skip to content

Commit e93c0be

Browse files
wannaphongwannaphong
committed
Close some test to save space
1 parent 9c9f881 commit e93c0be

File tree

6 files changed

+69
-81
lines changed

6 files changed

+69
-81
lines changed

‎tests/test_augment.py‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33
import unittest
44
from pythainlp.augment import WordNetAug
55
from pythainlp.augment.wordnet import postype2wordnet
6-
from pythainlp.augment.lm import Thai2transformersAug
6+
# from pythainlp.augment.lm import Thai2transformersAug
77
from pythainlp.augment.word2vec.bpemb_wv import BPEmbAug
88
from pythainlp.augment.word2vec import (
9-
Thai2fitAug,
109
LTW2VAug
1110
)
11+
# Thai2fitAug,
1212
import nltk
1313

1414

@@ -26,10 +26,10 @@ def test_WordNetAug(self):
2626
self.assertIsNone(postype2wordnet('n', 'abc'))
2727
self.assertIsNotNone(postype2wordnet('NOUN', 'orchid'))
2828

29-
def test_Thai2fitAug(self):
30-
_aug = Thai2fitAug()
31-
self.assertIsNotNone(_aug.tokenizer(self.text))
32-
self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
29+
# def test_Thai2fitAug(self):
30+
# _aug = Thai2fitAug()
31+
# self.assertIsNotNone(_aug.tokenizer(self.text))
32+
# self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
3333

3434
def test_BPEmbAug(self):
3535
_aug = BPEmbAug()
@@ -41,6 +41,6 @@ def test_LTW2VAug(self):
4141
self.assertIsNotNone(_aug.tokenizer(self.text))
4242
self.assertIsNotNone(_aug.augment(self.text, n_sent=3, p=0.5))
4343

44-
def test_Thai2transformersAug(self):
45-
_aug = Thai2transformersAug()
46-
self.assertIsNotNone(_aug.augment(self.text2, num_replace_tokens=1))
44+
# def test_Thai2transformersAug(self):
45+
# _aug = Thai2transformersAug()
46+
# self.assertIsNotNone(_aug.augment(self.text2, num_replace_tokens=1))

‎tests/test_generate.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def test_trigram(self):
4040
self.assertIsNotNone(_trigram.gen_sentence())
4141
self.assertIsNotNone(_trigram.gen_sentence(duplicate=True))
4242

43-
def test_thai2fit(self):
44-
self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว"))
45-
self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว", output_str=False))
46-
self.assertIsNotNone(gen_sentence())
43+
# def test_thai2fit(self):
44+
# self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว"))
45+
# self.assertIsNotNone(gen_sentence("กาลครั้งหนึ่งนานมาแล้ว", output_str=False))
46+
# self.assertIsNotNone(gen_sentence())

‎tests/test_parse.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@ class TestParsePackage(unittest.TestCase):
88
def test_dependency_parsing(self):
99
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="esupar"))
1010
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="esupar", tag="list"))
11-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud"))
12-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud", tag="list"))
13-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai"))
14-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai", tag="list"))
15-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith"))
16-
self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith", tag="list"))
11+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud"))
12+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="transformers_ud", tag="list"))
13+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai"))
14+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="spacy_thai", tag="list"))
15+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith"))
16+
# self.assertIsNotNone(dependency_parsing("ผมเป็นคนดี", engine="ud_goeswith", tag="list"))

‎tests/test_summarize.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,9 @@ def test_summarize(self):
2020
summarize(text=text, n=1),
2121
["อาหารจะต้องไม่มีพิษและไม่เกิดโทษต่อร่างกาย"],
2222
)
23-
self.assertIsNotNone(summarize(text, engine="mt5-small"))
24-
self.assertIsNotNone(summarize([]))
25-
self.assertIsNotNone(summarize(text, 1, engine="mt5-small"))
23+
# self.assertIsNotNone(summarize(text, engine="mt5-small"))
24+
# self.assertIsNotNone(summarize([]))
25+
# self.assertIsNotNone(summarize(text, 1, engine="mt5-small"))
2626
self.assertIsNotNone(summarize(text, 1, engine="mt5-cpe-kmutt-thai-sentence-sum"))
2727
self.assertIsNotNone(summarize(text, 1, engine="XX"))
2828
with self.assertRaises(ValueError):

‎tests/test_tag.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -346,10 +346,10 @@ def test_NER_class(self):
346346
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
347347
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
348348
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
349-
ner = NER(engine="wangchanberta")
350-
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
351-
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
352-
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
349+
# ner = NER(engine="wangchanberta")
350+
# self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
351+
# self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))
352+
# self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", tag=True))
353353
ner = NER(engine="thainer-v2")
354354
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า"))
355355
self.assertIsNotNone(ner.tag("แมวทำอะไรตอนห้าโมงเช้า", pos=False))

‎tests/test_translate.py‎

Lines changed: 43 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
class TestTranslatePackage(unittest.TestCase):
1919
def test_translate(self):
20-
remove("scb_1m_th-en_spm")
20+
# remove("scb_1m_th-en_spm")
2121
self.assertIsNone(download_model_all())
2222
self.th_en_translator = ThEnTranslator()
2323
self.assertIsNotNone(
@@ -31,59 +31,47 @@ def test_translate(self):
3131
"the cat eats fish.",
3232
)
3333
)
34-
self.th_zh_translator = ThZhTranslator()
35-
self.assertIsNotNone(
36-
self.th_zh_translator.translate(
37-
"ผมรักคุณ",
38-
)
39-
)
40-
self.zh_th_translator = ZhThTranslator()
41-
self.assertIsNotNone(
42-
self.zh_th_translator.translate(
43-
"我爱你",
44-
)
45-
)
46-
self.th_en_translator = Translate('th', 'en')
47-
self.assertIsNotNone(
48-
self.th_en_translator.translate(
49-
"แมวกินปลา",
50-
)
51-
)
52-
self.en_th_translator = Translate('en', 'th')
53-
self.assertIsNotNone(
54-
self.en_th_translator.translate(
55-
"the cat eats fish.",
56-
)
57-
)
58-
self.th_zh_translator = Translate('th', 'zh')
59-
self.assertIsNotNone(
60-
self.th_zh_translator.translate(
61-
"ผมรักคุณ",
62-
)
63-
)
64-
self.zh_th_translator = Translate('zh', 'th')
65-
self.assertIsNotNone(
66-
self.zh_th_translator.translate(
67-
"我爱你",
68-
)
69-
)
70-
self.th_fr_translator = Translate('th', 'fr')
71-
self.assertIsNotNone(
72-
self.th_fr_translator.translate(
73-
"ทดสอบระบบ",
74-
)
75-
)
76-
self.th_fr_translator = Translate('th', 'fr', engine="small100")
77-
self.assertIsNotNone(
78-
self.th_fr_translator.translate(
79-
"ทดสอบระบบ",
80-
)
81-
)
82-
self.th_ja_translator = Translate('th', 'ja', engine="small100")
83-
self.assertIsNotNone(
84-
self.th_fr_translator.translate(
85-
"ทดสอบระบบ",
86-
)
87-
)
34+
# self.th_zh_translator = ThZhTranslator()
35+
# self.assertIsNotNone(
36+
# self.th_zh_translator.translate(
37+
# "ผมรักคุณ",
38+
# )
39+
# )
40+
# self.zh_th_translator = ZhThTranslator()
41+
# self.assertIsNotNone(
42+
# self.zh_th_translator.translate(
43+
# "我爱你",
44+
# )
45+
# )
46+
# self.th_zh_translator = Translate('th', 'zh')
47+
# self.assertIsNotNone(
48+
# self.th_zh_translator.translate(
49+
# "ผมรักคุณ",
50+
# )
51+
# )
52+
# self.zh_th_translator = Translate('zh', 'th')
53+
# self.assertIsNotNone(
54+
# self.zh_th_translator.translate(
55+
# "我爱你",
56+
# )
57+
# )
58+
# self.th_fr_translator = Translate('th', 'fr')
59+
# self.assertIsNotNone(
60+
# self.th_fr_translator.translate(
61+
# "ทดสอบระบบ",
62+
# )
63+
# )
64+
# self.th_fr_translator = Translate('th', 'fr', engine="small100")
65+
# self.assertIsNotNone(
66+
# self.th_fr_translator.translate(
67+
# "ทดสอบระบบ",
68+
# )
69+
# )
70+
# self.th_ja_translator = Translate('th', 'ja', engine="small100")
71+
# self.assertIsNotNone(
72+
# self.th_fr_translator.translate(
73+
# "ทดสอบระบบ",
74+
# )
75+
# )
8876
with self.assertRaises(ValueError):
8977
self.th_cat_translator = Translate('th', 'cat', engine="fkfj")

0 commit comments

Comments
 (0)