/
testing.go
71 lines (60 loc) · 2.79 KB
/
testing.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package flatrtree
import (
"fmt"
"testing"
"github.com/stretchr/testify/require"
)
var testDegrees = []int{5, DefaultDegree, 15}
var testBuilders = map[string]func() Builder{
"Hilbert": func() Builder { return NewHilbertBuilder() },
"OMT": func() Builder { return NewOMTBuilder() },
}
type rtreeTestCase struct {
name string
index *RTree
items []float64
count int
degree int
}
func createTestCases(t *testing.T) []rtreeTestCase {
var result []rtreeTestCase
for builderName, newBuilder := range testBuilders {
for _, degree := range testDegrees {
for _, count := range []int{0, 1, degree, 100} {
name := fmt.Sprintf("%v/deg=%d/count=%d", builderName, degree, count)
index, items := createIndex(t, newBuilder, count, degree)
result = append(result, rtreeTestCase{name, index, items, count, degree})
}
}
}
return result
}
func createIndex(t *testing.T, newBuilder func() Builder, count int, degree int) (*RTree, []float64) {
var testBoxes = []float64{
8, 62, 11, 66, 57, 17, 57, 19, 76, 26, 79, 29, 36, 56, 38, 56, 92, 77, 96, 80, 87, 70, 90, 74,
43, 41, 47, 43, 0, 58, 2, 62, 76, 86, 80, 89, 27, 13, 27, 15, 71, 63, 75, 67, 25, 2, 27, 2, 87,
6, 88, 6, 22, 90, 23, 93, 22, 89, 22, 93, 57, 11, 61, 13, 61, 55, 63, 56, 17, 85, 21, 87, 33,
43, 37, 43, 6, 1, 7, 3, 80, 87, 80, 87, 23, 50, 26, 52, 58, 89, 58, 89, 12, 30, 15, 34, 32, 58,
36, 61, 41, 84, 44, 87, 44, 18, 44, 19, 13, 63, 15, 67, 52, 70, 54, 74, 57, 59, 58, 59, 17, 90,
20, 92, 48, 53, 52, 56, 92, 68, 92, 72, 26, 52, 30, 52, 56, 23, 57, 26, 88, 48, 88, 48, 66, 13,
67, 15, 7, 82, 8, 86, 46, 68, 50, 68, 37, 33, 38, 36, 6, 15, 8, 18, 85, 36, 89, 38, 82, 45, 84,
48, 12, 2, 16, 3, 26, 15, 26, 16, 55, 23, 59, 26, 76, 37, 79, 39, 86, 74, 90, 77, 16, 75, 18,
78, 44, 18, 45, 21, 52, 67, 54, 71, 59, 78, 62, 78, 24, 5, 24, 8, 64, 80, 64, 83, 66, 55, 70,
55, 0, 17, 2, 19, 15, 71, 18, 74, 87, 57, 87, 59, 6, 34, 7, 37, 34, 30, 37, 32, 51, 19, 53, 19,
72, 51, 73, 55, 29, 45, 30, 45, 94, 94, 96, 95, 7, 22, 11, 24, 86, 45, 87, 48, 33, 62, 34, 65,
18, 10, 21, 14, 64, 66, 67, 67, 64, 25, 65, 28, 27, 4, 31, 6, 84, 4, 85, 5, 48, 80, 50, 81, 1,
61, 3, 61, 71, 89, 74, 92, 40, 42, 43, 43, 27, 64, 28, 66, 46, 26, 50, 26, 53, 83, 57, 87, 14,
75, 15, 79, 31, 45, 34, 45, 89, 84, 92, 88, 84, 51, 85, 53, 67, 87, 67, 89, 39, 26, 43, 27, 47,
61, 47, 63, 23, 49, 25, 53, 12, 3, 14, 5, 16, 50, 19, 53, 63, 80, 64, 84, 22, 63, 22, 64, 26,
66, 29, 66, 2, 15, 3, 15, 74, 77, 77, 79, 64, 11, 68, 11, 38, 4, 39, 8, 83, 73, 87, 77, 85, 52,
89, 56, 74, 60, 76, 63, 62, 66, 65, 67,
}
items := testBoxes[:count*4]
builder := newBuilder()
for i := 0; i < len(items)/4; i++ {
builder.Add(int64(i), items[i*4], items[i*4+1], items[i*4+2], items[i*4+3])
}
index, err := builder.Finish(degree)
require.Nil(t, err)
return index, items
}