add more test code

go-ego · Jun 30, 2020 · 211f676 · 211f676
1 parent 17af0e0
commit 211f676
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/phrase/paragraph_test.go b/phrase/paragraph_test.go
@@ -39,8 +39,10 @@ func TestParagraph(t *testing.T) {
 }
 
 func TestPinyin(t *testing.T) {
-	seg := gse.New()
+	seg := gse.New("zh, ../examples/dict.txt")
 	WithGse(seg)
-	p := Pinyin("西雅图太空针")
-	tt.Equal(t, "[xi ya tu tai kong zhen]", p)
+
+	AddDict("都会区", "dū huì qū")
+	p := Pinyin("西雅图都会区, 西雅图太空针")
+	tt.Equal(t, "[xi ya tu du hui qu, xi ya tu tai kong zhen]", p)
 }