focus/tree/tried/tried_test.go

119 lines
2.0 KiB
Go
Raw Normal View History

package tried
import (
"testing"
2019-08-19 06:30:24 +00:00
"github.com/Pallinder/go-randomdata"
)
func TestTried_PutAndGet1(t *testing.T) {
tried := New()
2019-08-19 06:30:24 +00:00
tried.Put(TriedString("asdf"))
tried.Put(TriedString("hehe"), "hehe")
tried.Put(TriedString("xixi"), 3)
var result interface{}
2019-08-19 06:30:24 +00:00
result = tried.Get(TriedString("asdf"))
if result != tried {
t.Error("result should be 3")
}
2019-08-19 06:30:24 +00:00
result = tried.Get(TriedString("xixi"))
if result != 3 {
t.Error("result should be 3")
}
2019-08-19 06:30:24 +00:00
result = tried.Get(TriedString("hehe"))
if result != "hehe" {
t.Error("result should be hehe")
}
2019-08-19 06:30:24 +00:00
result = tried.Get(TriedString("haha"))
if result != nil {
t.Error("result should be nil")
}
2019-08-19 06:30:24 +00:00
result = tried.Get(TriedString("b"))
if result != nil {
t.Error("result should be nil")
}
}
func TestTried_Traversal(t *testing.T) {
tried := New()
2019-08-19 06:30:24 +00:00
tried.Put(TriedString("asdf"))
tried.Put(TriedString("abdf"), "ab")
tried.Put(TriedString("hehe"), "hehe")
tried.Put(TriedString("xixi"), 3)
var result []interface{}
tried.Traversal(func(idx uint, v interface{}) bool {
// t.Error(idx, v)
result = append(result, v)
return true
})
if result[0] != "ab" {
t.Error(result[0])
}
if result[1] != tried {
t.Error(result[1])
}
if result[2] != "hehe" {
t.Error(result[2])
}
if result[3] != 3 {
t.Error(result[3])
}
2019-08-19 06:30:24 +00:00
}
func BenchmarkTried_Put(b *testing.B) {
var data []TriedString
b.N = 10000
count := 1000
for i := 0; i < b.N; i++ {
data = append(data, TriedString(randomdata.RandStringRunes(10)+randomdata.RandStringRunes(4)))
}
b.ResetTimer()
b.N = b.N * count
for c := 0; c < count; c++ {
tried := New()
for _, v := range data {
tried.Put(v)
}
}
}
2019-08-19 06:30:24 +00:00
func BenchmarkTried_Get(b *testing.B) {
var data []TriedString
b.N = 10000
count := 1000
for i := 0; i < b.N; i++ {
data = append(data, TriedString(randomdata.RandStringRunes(10)+randomdata.RandStringRunes(4)))
}
b.N = b.N * count
tried := New()
for _, v := range data {
tried.Put(v)
}
b.ResetTimer()
for c := 0; c < count; c++ {
for _, v := range data {
tried.Get(v)
}
}
}