202 lines
3.3 KiB
Go
202 lines
3.3 KiB
Go
package heap
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/474420502/focus/compare"
|
|
)
|
|
|
|
func TestHeapPushTopPop(t *testing.T) {
|
|
h := New(compare.Int)
|
|
l := []int{9, 5, 15, 2, 3}
|
|
ol := []int{15, 9, 5, 3, 2}
|
|
for _, v := range l {
|
|
h.Put(v)
|
|
}
|
|
|
|
for _, tv := range ol {
|
|
if v, isfound := h.Top(); isfound {
|
|
if !(isfound && v == tv) {
|
|
t.Error(v)
|
|
}
|
|
}
|
|
|
|
if v, isfound := h.Pop(); isfound {
|
|
if !(isfound && v == tv) {
|
|
t.Error(v)
|
|
}
|
|
}
|
|
}
|
|
|
|
if h.Size() != 0 {
|
|
t.Error("heap size is not equals to zero")
|
|
}
|
|
}
|
|
|
|
// func Int(k1, k2 interface{}) int {
|
|
// c1 := k1.(int)
|
|
// c2 := k2.(int)
|
|
// switch {
|
|
// case c1 > c2:
|
|
// return -1
|
|
// case c1 < c2:
|
|
// return 1
|
|
// default:
|
|
// return 0
|
|
// }
|
|
// }
|
|
|
|
// func TestPush(t *testing.T) {
|
|
|
|
// for i := 0; i < 1000000; i++ {
|
|
// h := New(Int)
|
|
|
|
// gods := binaryheap.NewWithIntComparator()
|
|
// for c := 0; c < 20; c++ {
|
|
// v := randomdata.Number(0, 100)
|
|
// h.Push(v)
|
|
// gods.Push(v)
|
|
// }
|
|
|
|
// r1 := spew.Sprint(h.Values())
|
|
// r2 := spew.Sprint(gods.Values())
|
|
// if r1 != r2 {
|
|
// t.Error(r1)
|
|
// t.Error(r2)
|
|
// break
|
|
// }
|
|
// }
|
|
|
|
// }
|
|
|
|
// func TestPop(t *testing.T) {
|
|
|
|
// for i := 0; i < 200000; i++ {
|
|
// h := New(Int)
|
|
|
|
// // m := make(map[int]int)
|
|
// gods := binaryheap.NewWithIntComparator()
|
|
// for c := 0; c < 40; c++ {
|
|
// v := randomdata.Number(0, 100)
|
|
// // if _, ok := m[v]; !ok {
|
|
// h.Push(v)
|
|
// gods.Push(v)
|
|
// // m[v] = v
|
|
// // }
|
|
|
|
// }
|
|
|
|
// // t.Error(h.Values())
|
|
// // t.Error(gods.Values())
|
|
// for c := 0; c < randomdata.Number(5, 10); c++ {
|
|
// v1, _ := h.Pop()
|
|
// v2, _ := gods.Pop()
|
|
|
|
// if v1 != v2 {
|
|
// t.Error(h.Values(), v1)
|
|
// t.Error(gods.Values(), v2)
|
|
// return
|
|
// }
|
|
// }
|
|
|
|
// r1 := spew.Sprint(h.Values())
|
|
// r2 := spew.Sprint(gods.Values())
|
|
// if r1 != r2 {
|
|
// t.Error(r1)
|
|
// t.Error(r2)
|
|
// break
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// func BenchmarkPush(b *testing.B) {
|
|
|
|
// l := loadTestData()
|
|
|
|
// b.ResetTimer()
|
|
// execCount := 50
|
|
// b.N = len(l) * execCount
|
|
|
|
// for c := 0; c < execCount; c++ {
|
|
// b.StopTimer()
|
|
// h := New(Int)
|
|
// b.StartTimer()
|
|
// for _, v := range l {
|
|
// h.Push(v)
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// func BenchmarkPop(b *testing.B) {
|
|
|
|
// h := New(Int)
|
|
|
|
// l := loadTestData()
|
|
|
|
// b.ResetTimer()
|
|
// execCount := 20
|
|
// b.N = len(l) * execCount
|
|
|
|
// for c := 0; c < execCount; c++ {
|
|
// b.StopTimer()
|
|
// for _, v := range l {
|
|
// h.Push(v)
|
|
// }
|
|
// b.StartTimer()
|
|
// for h.size != 0 {
|
|
// h.Pop()
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// func BenchmarkGodsPop(b *testing.B) {
|
|
|
|
// h := binaryheap.NewWithIntComparator()
|
|
|
|
// l := loadTestData()
|
|
|
|
// b.ResetTimer()
|
|
// execCount := 20
|
|
// b.N = len(l) * execCount
|
|
|
|
// for c := 0; c < execCount; c++ {
|
|
// b.StopTimer()
|
|
// for _, v := range l {
|
|
// h.Push(v)
|
|
// }
|
|
// b.StartTimer()
|
|
// for h.Size() != 0 {
|
|
// h.Pop()
|
|
// }
|
|
// }
|
|
|
|
// }
|
|
|
|
// func BenchmarkGodsPush(b *testing.B) {
|
|
// l := loadTestData()
|
|
|
|
// b.ResetTimer()
|
|
// execCount := 50
|
|
// b.N = len(l) * execCount
|
|
|
|
// for c := 0; c < execCount; c++ {
|
|
// b.StopTimer()
|
|
// h := binaryheap.NewWith(Int)
|
|
// b.StartTimer()
|
|
// for _, v := range l {
|
|
// h.Push(v)
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// func loadTestData() []int {
|
|
// data, err := ioutil.ReadFile("../l.log")
|
|
// if err != nil {
|
|
// log.Println(err)
|
|
// }
|
|
// var l []int
|
|
// decoder := gob.NewDecoder(bytes.NewReader(data))
|
|
// decoder.Decode(&l)
|
|
// return l
|
|
// }
|