1
0
mirror of https://github.com/fumiama/jieba.git synced 2026-06-05 00:32:51 +08:00
Files
jieba/posseg/viterbi_test.go
源文雨 8bbc755ed4 优化
2022-11-30 12:18:15 +08:00

72 lines
1.8 KiB
Go
Executable File

package posseg
import (
"testing"
)
var defaultRoute []tag
func init() {
var t tag
t, _ = newTag("B", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("M", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("E", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("S", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("B", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("E", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("B", "n")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("M", "n")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("E", "n")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("S", "d")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("S", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("S", "n")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("B", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("E", "v")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("B", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("M", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("M", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("M", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("E", "nr")
defaultRoute = append(defaultRoute, t)
t, _ = newTag("S", "zg")
defaultRoute = append(defaultRoute, t)
}
func TestViterbi(t *testing.T) {
ss := "李小福是创新办主任也是云计算方面的专家;"
route := viterbi([]rune(ss))
if len(route) != len(defaultRoute) {
t.Fatal(len(route))
}
for index := range route {
if route[index] != defaultRoute[index] {
t.Fatal(route[index])
}
}
}
func BenchmarkViterbi(b *testing.B) {
ss := "李小福是创新办主任也是云计算方面的专家;"
for i := 0; i < b.N; i++ {
viterbi([]rune(ss))
}
}