OLD | NEW |
(Empty) | |
| 1 # 2010 August 28 |
| 2 # |
| 3 # The author disclaims copyright to this source code. In place of |
| 4 # a legal notice, here is a blessing: |
| 5 # |
| 6 # May you do good and not evil. |
| 7 # May you find forgiveness for yourself and forgive others. |
| 8 # May you share freely, never taking more than you give. |
| 9 # |
| 10 #*********************************************************************** |
| 11 # This file contains tests for the r-tree module. Specifically, it tests |
| 12 # that new-style custom r-tree queries (geometry callbacks) work. |
| 13 # |
| 14 |
| 15 if {![info exists testdir]} { |
| 16 set testdir [file join [file dirname [info script]] .. .. test] |
| 17 } |
| 18 source $testdir/tester.tcl |
| 19 ifcapable !rtree { finish_test ; return } |
| 20 ifcapable rtree_int_only { finish_test; return } |
| 21 |
| 22 |
| 23 #------------------------------------------------------------------------- |
| 24 # Test the example 2d "circle" geometry callback. |
| 25 # |
| 26 register_circle_geom db |
| 27 |
| 28 do_execsql_test rtreeE-1.1 { |
| 29 PRAGMA page_size=512; |
| 30 CREATE VIRTUAL TABLE rt1 USING rtree(id,x0,x1,y0,y1); |
| 31 |
| 32 /* A tight pattern of small boxes near 0,0 */ |
| 33 WITH RECURSIVE |
| 34 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), |
| 35 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) |
| 36 INSERT INTO rt1 SELECT x+5*y, x, x+2, y, y+2 FROM x, y; |
| 37 |
| 38 /* A looser pattern of small boxes near 100, 0 */ |
| 39 WITH RECURSIVE |
| 40 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), |
| 41 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) |
| 42 INSERT INTO rt1 SELECT 100+x+5*y, x*3+100, x*3+102, y*3, y*3+2 FROM x, y; |
| 43 |
| 44 /* A looser pattern of larger boxes near 0, 200 */ |
| 45 WITH RECURSIVE |
| 46 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), |
| 47 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) |
| 48 INSERT INTO rt1 SELECT 200+x+5*y, x*7, x*7+15, y*7+200, y*7+215 FROM x, y; |
| 49 } {} |
| 50 |
| 51 # Queries against each of the three clusters */ |
| 52 do_execsql_test rtreeE-1.1 { |
| 53 SELECT id FROM rt1 WHERE id MATCH Qcircle(0.0, 0.0, 50.0, 3) ORDER BY id; |
| 54 } {0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24} |
| 55 do_execsql_test rtreeE-1.2 { |
| 56 SELECT id FROM rt1 WHERE id MATCH Qcircle(100.0, 0.0, 50.0, 3) ORDER BY id; |
| 57 } {100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 1
19 120 121 122 123 124} |
| 58 do_execsql_test rtreeE-1.3 { |
| 59 SELECT id FROM rt1 WHERE id MATCH Qcircle(0.0, 200.0, 50.0, 3) ORDER BY id; |
| 60 } {200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 2
19 220 221 222 223 224} |
| 61 |
| 62 # The Qcircle geometry function gives a lower score to larger leaf-nodes. |
| 63 # This causes the 200s to sort before the 100s and the 0s to sort before |
| 64 # last. |
| 65 # |
| 66 do_execsql_test rtreeE-1.4 { |
| 67 SELECT id FROM rt1 WHERE id MATCH Qcircle(0,0,1000,3) AND id%100==0 |
| 68 } {200 100 0} |
| 69 |
| 70 # Exclude odd rowids on a depth-first search |
| 71 do_execsql_test rtreeE-1.5 { |
| 72 SELECT id FROM rt1 WHERE id MATCH Qcircle(0,0,1000,4) ORDER BY +id |
| 73 } {0 2 4 6 8 10 12 14 16 18 20 22 24 100 102 104 106 108 110 112 114 116 118 120
122 124 200 202 204 206 208 210 212 214 216 218 220 222 224} |
| 74 |
| 75 # Exclude odd rowids on a breadth-first search. |
| 76 do_execsql_test rtreeE-1.6 { |
| 77 SELECT id FROM rt1 WHERE id MATCH Qcircle(0,0,1000,5) ORDER BY +id |
| 78 } {0 2 4 6 8 10 12 14 16 18 20 22 24 100 102 104 106 108 110 112 114 116 118 120
122 124 200 202 204 206 208 210 212 214 216 218 220 222 224} |
| 79 |
| 80 # Construct a large 2-D RTree with thousands of random entries. |
| 81 # |
| 82 do_test rtreeE-2.1 { |
| 83 db eval { |
| 84 CREATE TABLE t2(id,x0,x1,y0,y1); |
| 85 CREATE VIRTUAL TABLE rt2 USING rtree(id,x0,x1,y0,y1); |
| 86 BEGIN; |
| 87 } |
| 88 expr srand(0) |
| 89 for {set i 1} {$i<=10000} {incr i} { |
| 90 set dx [expr {int(rand()*40)+1}] |
| 91 set dy [expr {int(rand()*40)+1}] |
| 92 set x0 [expr {int(rand()*(10000 - $dx))}] |
| 93 set x1 [expr {$x0+$dx}] |
| 94 set y0 [expr {int(rand()*(10000 - $dy))}] |
| 95 set y1 [expr {$y0+$dy}] |
| 96 set id [expr {$i+10000}] |
| 97 db eval {INSERT INTO t2 VALUES($id,$x0,$x1,$y0,$y1)} |
| 98 } |
| 99 db eval { |
| 100 INSERT INTO rt2 SELECT * FROM t2; |
| 101 COMMIT; |
| 102 } |
| 103 } {} |
| 104 |
| 105 for {set i 1} {$i<=200} {incr i} { |
| 106 set dx [expr {int(rand()*100)}] |
| 107 set dy [expr {int(rand()*100)}] |
| 108 set x0 [expr {int(rand()*(10000 - $dx))}] |
| 109 set x1 [expr {$x0+$dx}] |
| 110 set y0 [expr {int(rand()*(10000 - $dy))}] |
| 111 set y1 [expr {$y0+$dy}] |
| 112 set ans [db eval {SELECT id FROM t2 WHERE x1>=$x0 AND x0<=$x1 AND y1>=$y0 AND
y0<=$y1 ORDER BY id}] |
| 113 do_execsql_test rtreeE-2.2.$i { |
| 114 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch($x0,$x1,$y0,$y1) ORDER
BY id |
| 115 } $ans |
| 116 } |
| 117 |
| 118 # Run query that have very deep priority queues |
| 119 # |
| 120 set ans [db eval {SELECT id FROM t2 WHERE x1>=0 AND x0<=5000 AND y1>=0 AND y0<=5
000 ORDER BY id}] |
| 121 do_execsql_test rtreeE-2.3 { |
| 122 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch(0,5000,0,5000) ORDER BY i
d |
| 123 } $ans |
| 124 set ans [db eval {SELECT id FROM t2 WHERE x1>=0 AND x0<=10000 AND y1>=0 AND y0<=
10000 ORDER BY id}] |
| 125 do_execsql_test rtreeE-2.4 { |
| 126 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch(0,10000,0,10000) ORDER BY
id |
| 127 } $ans |
| 128 |
| 129 finish_test |
OLD | NEW |