Added rtree test script.

This commit is contained in:
jpcaram 2014-12-27 17:26:00 -05:00
parent f52dee3a45
commit f528a07751
1 changed files with 24 additions and 0 deletions

24
tests/test_rt.py Normal file
View File

@ -0,0 +1,24 @@
from rtree import index as rtindex
def pt2rect(pt):
return pt[0], pt[1], pt[0], pt[1]
pts = [(0.0, 0.0), (1.0, 1.0), (0.0, 1.0)]
#p = rtindex.Property()
#p.buffering_capacity = 1
#rt = rtindex.Index(properties=p)
rt = rtindex.Index()
# If interleaved is True, the coordinates must be in
# the form [xmin, ymin, ..., kmin, xmax, ymax, ..., kmax].
print rt.interleaved
[rt.add(0, pt2rect(pt)) for pt in pts]
print [r.bbox for r in list(rt.nearest((0, 0), 10, True))]
for pt in pts:
rt.delete(0, pt2rect(pt))
print pt2rect(pt), [r.bbox for r in list(rt.nearest((0, 0), 10, True))]