Ejemplo n.º 1
0
func main() {
	t, err := btree.New(50) // branching factor of the btree. For high performance 100 is optimal.
	if err != nil {
		panic(err)
	}
	t.Insert(Int(8)) // Btree accepts btee.Comper interface
	fmt.Println(
		t.Find(Int(8)),
		t.Delete(Int(8)),
		t.Find(Int(8)),
	)
}
Ejemplo n.º 2
0
func TestSmallOrder() {
	for i := 5; i < 10; i++ {
		s, _ := btree.New(i)
		for j := 0; j < testnum/10; j++ {
			s.Insert(btree.Int(j))
		}
		for j := 0; j < testnum/10; j++ {
			if s.Find(btree.Int(j)) == false {
				panic("Can't find a value.")
			}
		}
		s2, _ := btree.New(i)
		for j := testnum / 10; j >= 0; j-- {
			s2.Insert(btree.Int(j))
		}
		for j := testnum / 10; j >= 0; j-- {
			if s2.Find(btree.Int(j)) == false {
				panic("Cant find a value.")
			}
		}
	}
}
Ejemplo n.º 3
0
func BenchmarkInsert() {
	order := 100
	p, _ := btree.New(order)
	fmt.Println("Measuring performance. Order is ", order, ", iteration count is ", perfnum, ":") // (1 million inserts in C++ version with ints only takes 0.1 secs if order is 100.)
	tim := time.Now()
	for i := 0; i < perfnum; i++ {
		p.Insert(btree.Int(i))
	}
	fmt.Println("Took ", time.Since(tim))
	fmt.Println("Measuring find of every value:")
	tim = time.Now()
	for i := 0; i < perfnum; i++ {
		p.Find(btree.Int(i))
	}
	fmt.Println("Took ", time.Since(tim))
}
Ejemplo n.º 4
0
func TestBtree() {
	order := 100
	// Small order test
	a, _ := btree.New(order)
	for i := 0; i < testnum; i++ {
		a.Insert(btree.Int(i))
	}
	for i := 0; i < testnum; i++ {
		if a.Find(btree.Int(i)) == false {
			fmt.Println(i)
			panic("Cant find value.")
		}
	}
	//arr := a.GetAll()
	//leafTest(a, arr)
	b, _ := btree.New(order)
	for i := testnum; i > 0; i-- {
		b.Insert(btree.Int(i))
	}
	for i := testnum; i > 0; i-- {
		if b.Find(btree.Int(i)) == false {
			fmt.Println(i)
			panic("Cant find value.")
		}
	}
	//leafTest(b, arr)
	if a.TreeSize() != testnum {
		panic("Tree \"a\" size is not correct.")
	}
	if b.TreeSize() != testnum {
		panic("Tree \"b\" size is not correct.")
	}
	u, _ := btree.New(order)
	for i := 0; i < testnum; i++ {
		u.Insert(btree.Int(i % 2))
	}
	for i := 0; i < 2; i++ {
		if u.Find(btree.Int(i)) == false {
			panic("Cant find value at unique test.")
		}
	}
	fmt.Println("Insert duplicates, then delete all of them and cry if any not found...")
	for ord := 5; ord < 43; ord++ {
		dup_inc, _ := btree.New(ord)
		dup_dec, _ := btree.New(ord)
		for mod := 2; mod < 47; mod++ {
			for k := 0; k <= testnum/5; k++ {
				dup_inc.Insert(btree.Int(k % mod))
			}
			for k := 0; k <= testnum/5; k++ {
				if dup_inc.Delete(btree.Int(k%mod)) == false {
					panic("Dupe test failed.")
				}
			}
			for k := testnum / 5; k >= 0; k-- {
				dup_dec.Insert(btree.Int(k % mod))
			}
			for k := testnum / 5; k >= 0; k-- {
				if dup_dec.Delete(btree.Int(k%mod)) == false {
					panic("Dupe test failed.")
				}
			}
		}
	}
	fmt.Println("Doing some slightly controlled stress test.")
	for ord := 5; ord <= 31; ord++ {
		m := make(map[int]int)
		tes, _ := btree.New(5)
		deletion_count := 0
		for i := 0; i <= tnum*200; i++ {
			c := rand.Int() % (tnum * 15)
			v, ok := m[c]
			if ok != tes.Find(btree.Int(c)) {
				fmt.Println(v, ok)
				panic("Deviation from map.")
			}
			if ok == true {
				m[c]++
			} else {
				m[c] = 1
			}
			tes.Insert(btree.Int(c))

			c = rand.Int() % (tnum * 15)
			v, ok = m[c]
			if ok != tes.Find(btree.Int(c)) {
				panic("Bad.")
			}
			if ok == true {
				if v == 1 {
					delete(m, c)
				} else {
					m[c]--
				}
				if tes.Delete(btree.Int(c)) == false {
					panic("Cant find value.")
				}
				deletion_count++
			}
		}
		fmt.Println("	Ran with order ", ord, " deleted ", deletion_count)
	}
	fmt.Println("Test ran successfully.")
}