focus/tree/tried/tried_test.go

package tried

import (
	"testing"

	"github.com/Pallinder/go-randomdata"
)

// func TestTried_PutAndGet1(t *testing.T) {
// 	tried := New()

// 	tried.Put(("asdf"))
// 	tried.Put(("hehe"), "hehe")
// 	tried.Put(("xixi"), 3)

// 	var result interface{}

// 	result = tried.Get("asdf")
// 	if result != tried {
// 		t.Error("result should be 3")
// 	}

// 	result = tried.Get("xixi")
// 	if result != 3 {
// 		t.Error("result should be 3")
// 	}

// 	result = tried.Get("hehe")
// 	if result != "hehe" {
// 		t.Error("result should be hehe")
// 	}

// 	result = tried.Get("haha")
// 	if result != nil {
// 		t.Error("result should be nil")
// 	}

// 	result = tried.Get("b")
// 	if result != nil {
// 		t.Error("result should be nil")
// 	}
// }

// func TestTried_Traversal(t *testing.T) {
// 	tried := New()
// 	tried.Put("asdf")
// 	tried.Put(("abdf"), "ab")
// 	tried.Put(("hehe"), "hehe")
// 	tried.Put(("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])
// 	}
// }

func BenchmarkTried_Put(b *testing.B) {

	var data []TriedString
	b.N = 1000000
	count := 10

	for i := 0; i < b.N; i++ {
		var content []rune
		for c := 0; c < randomdata.Number(5, 15); c++ {
			char := randomdata.Number(0, 26) + 'a'
			content = append(content, rune(byte(char)))
		}
		data = append(data, TriedString(string(content)))
	}

	b.ResetTimer()
	b.N = b.N * count
	for c := 0; c < count; c++ {
		tried := New()
		for _, v := range data {
			tried.Put(v)
		}
	}
}

func BenchmarkTried_Get(b *testing.B) {

	var data []TriedString
	b.N = 1000000
	count := 10

	for i := 0; i < b.N; i++ {
		var content []rune
		for c := 0; c < randomdata.Number(5, 15); c++ {
			char := randomdata.Number(0, 26) + 'a'
			content = append(content, rune(byte(char)))
		}
		data = append(data, TriedString(content))
	}

	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)
		}
	}
}