32 lines
453 B
Go
32 lines
453 B
Go
|
package tagengine
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNGramLength(t *testing.T) {
|
||
|
|
type Case struct {
|
||
|
|
Input string
|
||
|
|
Length int
|
||
|
|
}
|
||
|
|
|
||
|
|
cases := []Case{
|
||
|
|
{"a b c", 3},
|
||
|
|
{" xyz\nlkj dflaj a", 4},
|
||
|
|
{"a", 1},
|
||
|
|
{" a", 1},
|
||
|
|
{"a", 1},
|
||
|
|
{" a\n", 1},
|
||
|
|
{" a ", 1},
|
||
|
|
{"\tx\ny\nz q ", 4},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tc := range cases {
|
||
|
|
length := ngramLength(tc.Input)
|
||
|
|
if length != tc.Length {
|
||
|
|
log.Fatalf("%s: %d != %d", tc.Input, length, tc.Length)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|