mirror of https://github.com/tidwall/tile38.git
86 lines
1.6 KiB
Go
86 lines
1.6 KiB
Go
|
package sstring
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/rand"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/tidwall/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestShared(t *testing.T) {
|
||
|
|
for i := -1; i < 10; i++ {
|
||
|
|
var str string
|
||
|
|
func() {
|
||
|
|
defer func() {
|
||
|
|
assert.Assert(recover().(string) == "string not found")
|
||
|
|
}()
|
||
|
|
str = Load(i)
|
||
|
|
}()
|
||
|
|
assert.Assert(str == "")
|
||
|
|
}
|
||
|
|
assert.Assert(Store("hello") == 0)
|
||
|
|
assert.Assert(Store("") == 1)
|
||
|
|
assert.Assert(Store("jello") == 2)
|
||
|
|
assert.Assert(Store("hello") == 0)
|
||
|
|
assert.Assert(Store("") == 1)
|
||
|
|
assert.Assert(Store("jello") == 2)
|
||
|
|
str := Load(0)
|
||
|
|
assert.Assert(str == "hello")
|
||
|
|
str = Load(1)
|
||
|
|
assert.Assert(str == "")
|
||
|
|
str = Load(2)
|
||
|
|
assert.Assert(str == "jello")
|
||
|
|
|
||
|
|
assert.Assert(Len() == 3)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func randStr(n int) string {
|
||
|
|
b := make([]byte, n)
|
||
|
|
rand.Read(b)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
b[i] = 'a' + b[i]%26
|
||
|
|
}
|
||
|
|
return string(b)
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkStore(b *testing.B) {
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
wmap := make(map[string]bool, b.N)
|
||
|
|
for len(wmap) < b.N {
|
||
|
|
wmap[randStr(10)] = true
|
||
|
|
}
|
||
|
|
words := make([]string, 0, b.N)
|
||
|
|
for word := range wmap {
|
||
|
|
words = append(words, word)
|
||
|
|
}
|
||
|
|
b.ResetTimer()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
Store(words[i])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkLoad(b *testing.B) {
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
wmap := make(map[string]bool, b.N)
|
||
|
|
for len(wmap) < b.N {
|
||
|
|
wmap[randStr(10)] = true
|
||
|
|
}
|
||
|
|
words := make([]string, 0, b.N)
|
||
|
|
for word := range wmap {
|
||
|
|
words = append(words, word)
|
||
|
|
}
|
||
|
|
var nums []int
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
nums = append(nums, Store(words[i]))
|
||
|
|
}
|
||
|
|
rand.Shuffle(len(nums), func(i, j int) {
|
||
|
|
nums[i], nums[j] = nums[j], nums[i]
|
||
|
|
})
|
||
|
|
b.ResetTimer()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
Load(nums[i])
|
||
|
|
}
|
||
|
|
}
|