@@ -13,7 +13,7 @@ func TestKiwiVersion(t *testing.T) {
1313}
1414
1515func TestAnalyze (t * testing.T ) {
16- kiwi := New ("./ModelGenerator " , 1 , KIWI_BUILD_DEFAULT )
16+ kiwi := New ("./base " , 1 , KIWI_BUILD_DEFAULT )
1717 res , _ := kiwi .Analyze ("아버지가 방에 들어가신다" , 1 , KIWI_MATCH_ALL )
1818
1919 expected := []TokenResult {
@@ -64,7 +64,7 @@ func TestAnalyze(t *testing.T) {
6464}
6565
6666func TestSplitSentence (t * testing.T ) {
67- kiwi := New ("./ModelGenerator " , 1 , KIWI_BUILD_DEFAULT )
67+ kiwi := New ("./base " , 1 , KIWI_BUILD_DEFAULT )
6868 res , _ := kiwi .SplitSentence ("여러 문장으로 구성된 텍스트네 이걸 분리해줘" , KIWI_MATCH_ALL )
6969
7070 expected := []SplitResult {
@@ -85,14 +85,16 @@ func TestSplitSentence(t *testing.T) {
8585}
8686
8787func TestAddWordFail (t * testing.T ) {
88- kb := NewBuilder ("./ModelGenerator " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
88+ kb := NewBuilder ("./base " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
8989 add := kb .AddWord ("아버지가" , "SKO" , 0 )
90- assert .Equal (t , 0 , add )
90+ assert .Equal (t , - 1 , add )
9191 assert .Equal (t , 0 , kb .Close ())
92+
93+ KiwiClearError ()
9294}
9395
9496func TestAddWord (t * testing.T ) {
95- kb := NewBuilder ("./ModelGenerator " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
97+ kb := NewBuilder ("./base " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
9698 add := kb .AddWord ("아버지가" , "NNG" , 0 )
9799
98100 assert .Equal (t , 0 , add )
@@ -132,12 +134,12 @@ func TestAddWord(t *testing.T) {
132134 Form : "시" ,
133135 },
134136 {
135- Position : 12 ,
137+ Position : 11 ,
136138 Tag : "EF" ,
137139 Form : "ᆫ다" ,
138140 },
139141 },
140- Score : - 36.959194 ,
142+ Score : - 32.80881 ,
141143 },
142144 }
143145
@@ -146,7 +148,7 @@ func TestAddWord(t *testing.T) {
146148}
147149
148150func TestLoadDict (t * testing.T ) {
149- kb := NewBuilder ("./ModelGenerator " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
151+ kb := NewBuilder ("./base " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
150152 add := kb .LoadDict ("./example/user_dict.tsv" )
151153
152154 assert .Equal (t , 1 , add )
@@ -191,12 +193,12 @@ func TestLoadDict(t *testing.T) {
191193 Form : "시" ,
192194 },
193195 {
194- Position : 12 ,
196+ Position : 11 ,
195197 Tag : "EF" ,
196198 Form : "ᆫ다" ,
197199 },
198200 },
199- Score : - 36.959194 ,
201+ Score : - 32.80881 ,
200202 },
201203 }
202204
@@ -205,7 +207,7 @@ func TestLoadDict(t *testing.T) {
205207}
206208
207209func TestLoadDict2 (t * testing.T ) {
208- kb := NewBuilder ("./ModelGenerator " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
210+ kb := NewBuilder ("./base " , 1 , KIWI_BUILD_INTEGRATE_ALLOMORPH )
209211 add := kb .LoadDict ("./example/user_dict2.tsv" )
210212
211213 assert .Equal (t , 3 , add )
@@ -236,7 +238,7 @@ func TestLoadDict2(t *testing.T) {
236238 Form : "들어가신다" ,
237239 },
238240 },
239- Score : - 13.669565 ,
241+ Score : - 12.538677 ,
240242 },
241243 }
242244
@@ -245,7 +247,7 @@ func TestLoadDict2(t *testing.T) {
245247}
246248
247249func TestExtractWord (t * testing.T ) {
248- kb := NewBuilder ("./ModelGenerator " , 0 , KIWI_BUILD_DEFAULT )
250+ kb := NewBuilder ("./base " , 1 , KIWI_BUILD_DEFAULT )
249251 rs := strings .NewReader (`2008년에는 애국가의 작곡자 안익태가 1930년대에 독일 유학 기간 중 친일 활동을 했다는 사실이 밝혀졌다. 이후 안익태가 나치 독일 하의
250252베를린에서 만주국 10주년 건국 기념음악회를 지휘하는 동영상까지 발굴되어 관련 학계나 사회에 큰 충격을 주었다. 안익태가 친일 행적을 한 바
251253있다는 빼도박도 못할 증거가 나왔으니까. 영상물의 '만주환상곡'에는 우리가 현재 알고있는 '한국환상곡'의 두 선율("무궁화 삼천리 나의 사랑아,
@@ -263,12 +265,18 @@ func TestExtractWord(t *testing.T) {
263265 POSScore : - 1.92593 ,
264266 Score : 0 ,
265267 },
268+ {
269+ Form : "익태" ,
270+ Freq : 4 ,
271+ POSScore : - 0.23702252 ,
272+ Score : 0 ,
273+ },
266274 }, wordInfos )
267275 assert .Equal (t , 0 , kb .Close ())
268276}
269277
270278func TestExtractWordwithFile (t * testing.T ) {
271- kb := NewBuilder ("./ModelGenerator " , 0 , KIWI_BUILD_DEFAULT )
279+ kb := NewBuilder ("./base " , 0 , KIWI_BUILD_DEFAULT ) // Use single thread for deterministic results
272280 file , _ := os .Open ("./example/test.txt" )
273281
274282 wordInfos , _ := kb .ExtractWords (file , 10 /*=minCnt*/ , 5 /*=maxWordLen*/ , 0.0 /*=minScore*/ , - 25.0 /*=posThreshold*/ )
0 commit comments