-
Notifications
You must be signed in to change notification settings - Fork 270
/
test_translate.py
71 lines (68 loc) · 2.27 KB
/
test_translate.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# -*- coding: utf-8 -*-
import unittest
from pythainlp.translate import Translate
from pythainlp.translate.en_th import (
EnThTranslator,
ThEnTranslator,
download_model_all
)
class TestTranslatePackage(unittest.TestCase):
def test_translate(self):
# remove("scb_1m_th-en_spm")
self.assertIsNone(download_model_all())
self.th_en_translator = ThEnTranslator()
self.assertIsNotNone(
self.th_en_translator.translate(
"แมวกินปลา",
)
)
self.en_th_translator = EnThTranslator()
self.assertIsNotNone(
self.en_th_translator.translate(
"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_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")