mirror of https://github.com/tidwall/rtred.git
97 lines
1.8 KiB
Go
97 lines
1.8 KiB
Go
package rtree
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type tRect []float64
|
|
|
|
func (r *tRect) Arr() []float64 {
|
|
return []float64(*r)
|
|
}
|
|
func (r *tRect) Rect() (min, max []float64) {
|
|
return r.Arr()[:len(r.Arr())/2], r.Arr()[len(r.Arr())/2:]
|
|
}
|
|
|
|
func (r *tRect) String() string {
|
|
min, max := r.Rect()
|
|
return fmt.Sprintf("%v,%v", min, max)
|
|
}
|
|
func tRandRect(dims int) *tRect {
|
|
if dims == -1 {
|
|
dims = rand.Int()%4 + 1
|
|
}
|
|
r := tRect(make([]float64, dims*2))
|
|
for j := 0; j < dims; j++ {
|
|
minf := rand.Float64()*200 - 100
|
|
maxf := rand.Float64()*200 - 100
|
|
if minf > maxf {
|
|
minf, maxf = maxf, minf
|
|
}
|
|
r[j] = minf
|
|
r[dims+j] = maxf
|
|
}
|
|
return &r
|
|
}
|
|
func TestRTree(t *testing.T) {
|
|
tr := New()
|
|
|
|
tr.Insert(&tRect{10, 10, 10, 10, 20, 20, 20, 20})
|
|
tr.Insert(&tRect{10, 10, 10, 20, 20, 20})
|
|
tr.Insert(&tRect{10, 10, 20, 20})
|
|
tr.Insert(&tRect{10, 20})
|
|
|
|
if tr.Count() != 4 {
|
|
t.Fatalf("expecting %v, got %v", 4, tr.Count())
|
|
}
|
|
|
|
var count int
|
|
tr.Search(&tRect{0, 0, 0, 100, 100, 5}, func(item Item) bool {
|
|
count++
|
|
return true
|
|
})
|
|
|
|
if count != 2 {
|
|
t.Fatalf("expecting %v, got %v", 2, count)
|
|
}
|
|
}
|
|
|
|
func TestInsert(t *testing.T) {
|
|
rand.Seed(time.Now().UnixNano())
|
|
n := 50000
|
|
tr := New()
|
|
var r2arr []*tRect
|
|
for i := 0; i < n; i++ {
|
|
r := tRandRect(-1)
|
|
if len(r.Arr()) == 4 {
|
|
r2arr = append(r2arr, r)
|
|
}
|
|
tr.Insert(r)
|
|
}
|
|
if tr.Count() != n {
|
|
t.Fatalf("expecting %v, got %v", n, tr.Count())
|
|
}
|
|
var count int
|
|
tr.Search(&tRect{-100, -100, -100, -100, 100, 100, 100, 100}, func(item Item) bool {
|
|
if len(item.(*tRect).Arr()) == 4 {
|
|
count++
|
|
}
|
|
return true
|
|
})
|
|
p := float64(count) / float64(n)
|
|
|
|
if p < .23 || p > .27 {
|
|
t.Fatalf("bad random range, expected between 0.24-0.26, got %v", p)
|
|
}
|
|
for _, r := range r2arr {
|
|
tr.Remove(r)
|
|
}
|
|
total := tr.Count() + count
|
|
if total != n {
|
|
t.Fatalf("expected %v, got %v", n, total)
|
|
}
|
|
}
|