Skip to content

Commit 48f67d2

Browse files
committed
Add more test
1 parent 5518eac commit 48f67d2

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

tests/test_tokenize.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -589,3 +589,6 @@ def test_sefr_cut(self):
589589
self.assertIsNotNone(
590590
sefr_cut.segment("ฉันรักภาษาไทยเพราะฉันเป็นคนไทย"),
591591
)
592+
self.assertIsNotNone(
593+
sefr_cut.segment("ฉันรักภาษาไทยเพราะฉันเป็นคนไทย", engine="tnhc"),
594+
)

tests/test_wangchanberta.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import unittest
44

55
from pythainlp.wangchanberta import ThaiNameTagger, pos_tag, segment
6+
from pythainlp.wangchanberta.postag import PosTagTransformers
67

78

89
class TestWangchanberta(unittest.TestCase):
@@ -88,3 +89,12 @@ def test_pos_tag_wangchanberta(self):
8889
self.assertIsNotNone(
8990
pos_tag("ทดสอบระบบ", grouped_word=False)
9091
)
92+
_pos = PosTagTransformers()
93+
self.assertIsNotNone(
94+
_pos.tag("ทดสอบระบบ", grouped_word=False)
95+
)
96+
self.assertIsNotNone(
97+
_pos.tag("ทดสอบระบบ", grouped_word=True)
98+
)
99+
with self.assertRaises(NotImplementedError):
100+
pos_tag("I คิด therefore I am ผ็ฎ์", grouped_word=True, corpus="cat")

0 commit comments

Comments
 (0)