gslin/go-ir-playground
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add test cases for bigram.

Browse Source
This commit is contained in:
Gea-Suan Lin
2024-02-16 20:54:28 +08:00
parent 2b1e514431
commit 55ad14e790
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/ngram/ngram_test.go
View File

@@ -7,6 +7,16 @@ import (
"github.com/gslin/go-ir-playground/internal/ngram" "github.com/gslin/go-ir-playground/internal/ngram"
) )
func testBigram(t *testing.T) {
a := ngram.Bigram("test")
assert.Equal(t, len(a), 1)
assert.Equal(t, a[0], "test")
a = ngram.Bigram("測試")
assert.Equal(t, len(a), 1)
assert.Equal(t, a[0], "測試")
}
func TestUnigram(t *testing.T) { func TestUnigram(t *testing.T) {
a := ngram.Unigram("test") a := ngram.Unigram("test")
assert.Equal(t, len(a), 1) assert.Equal(t, len(a), 1)