20 lines
374 B
Go
20 lines
374 B
Go
package ngram_test
|
|
|
|
import (
|
|
"testing"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/gslin/go-ir-playground/internal/ngram"
|
|
)
|
|
|
|
func TestUnigram(t *testing.T) {
|
|
a := ngram.Unigram("test")
|
|
assert.Equal(t, len(a), 1)
|
|
assert.Equal(t, a[0], "test")
|
|
|
|
a = ngram.Unigram("測試")
|
|
assert.Equal(t, len(a), 2)
|
|
assert.Equal(t, a[0], "測")
|
|
assert.Equal(t, a[1], "試")
|
|
}
|