class GeometricTests(TestCase):
def setUp(self):
self.coords = set((sin(0.1*t), sin(0.2*t), sin(0.3*t)) for t in xrange(50))
self.o = Octree(((-1.0, 1.0),(-1.0, 1.0),(-1.0, 1.0)))
self.o.extend((p,None) for p in self.coords)
def test_basic(self):
self.assertEqual(len(self.o), 50)
def test_by_distance_from_point(self):
p = (0.123, 0.456, 0.789)
l1 = list(self.o.by_distance_from_point(p))
l2 = list(self.o.by_distance_from_point_rev(p))
self.assertEqual(set(c for (_,c,_) in l1), self.coords, "the points in order of distance from p should be the same as the points we put in")
for (d,c,v) in l1:
self.assertEqual(d, euclidean_point_point(p,c), "the points in order of distance from p should have distance computed correctly")
self.assertEqual(l1,list(reversed(l2)), "the points in order of distance reversed should be the reverse of the points in order of distance")
for ((d1,_,_),(d2,_,_)) in zip(l1,l1[1:]):
self.assertTrue(d1 <= d2, "the points in order of distance from p should have increasing distance from p")
self.assertEqual(self.o.nearest_to_point(p), l1[0], "the nearest point to p should be the first in order of distance from p")
l3 = list(self.o.near_point(p, 1.3))
self.assertEqual(l3, l1[:len(l3)], "the points near p should be an initial segment of the points in order of distance from p")
def test_embiggen(self):
b = ((-1.0,1.6),(-1.0,1.6),(-1.0,1.6))
o2 = self.o.rebound(b)
self.assertEqual(o2.bounds, b)
p = (0.653, -0.234, 0.113)
l1 = list(self.o.by_distance_from_point(p))
l2 = list(o2.by_distance_from_point(p))
self.assertEqual(l1, l2, "enlarging the bounds shouldn't have changed the points")
def test_restrict(self):
b = ((-1.57,0.43),(-0.76,0.83),(-0.37,1.96))
o2 = self.o.rebound(b)
self.assertEqual(o2.bounds, b)
p = (0.653, -0.234, 0.113)
l1 = [(d,c,v) for (d,c,v) in self.o.by_distance_from_point(p) if point_in_box(c,b)]
l2 = list(o2.by_distance_from_point(p))
self.assertEqual(l1, l2, "playing around with the bounds should restrict the points")
def test_remove_all(self):
p = (-0.432, 0.651, 0.791)
l = list(self.o.by_distance_from_point(p))
n = len(l)
for (i,(_,c,_)) in enumerate(l):
self.o.remove(c)
self.assertEqual(len(self.o), n-i-1)
with self.assertRaises(KeyError):
self.o.remove(c)
class GeometricTests(TestCase):
def setUp(self):
self.coords = set(
(sin(0.1 * t), sin(0.2 * t), sin(0.3 * t)) for t in xrange(50))
self.o = Octree(((-1.0, 1.0), (-1.0, 1.0), (-1.0, 1.0)))
self.o.extend((p, None) for p in self.coords)
def test_basic(self):
self.assertEqual(len(self.o), 50)
def test_by_distance_from_point(self):
p = (0.123, 0.456, 0.789)
l1 = list(self.o.by_distance_from_point(p))
l2 = list(self.o.by_distance_from_point_rev(p))
self.assertEqual(
set(c for (_, c, _) in l1), self.coords,
"the points in order of distance from p should be the same as the points we put in"
)
for (d, c, v) in l1:
self.assertEqual(
d, euclidean_point_point(p, c),
"the points in order of distance from p should have distance computed correctly"
)
self.assertEqual(
l1, list(reversed(l2)),
"the points in order of distance reversed should be the reverse of the points in order of distance"
)
for ((d1, _, _), (d2, _, _)) in zip(l1, l1[1:]):
self.assertTrue(
d1 <= d2,
"the points in order of distance from p should have increasing distance from p"
)
self.assertEqual(
self.o.nearest_to_point(p), l1[0],
"the nearest point to p should be the first in order of distance from p"
)
l3 = list(self.o.near_point(p, 1.3))
self.assertEqual(
l3, l1[:len(l3)],
"the points near p should be an initial segment of the points in order of distance from p"
)
def test_embiggen(self):
b = ((-1.0, 1.6), (-1.0, 1.6), (-1.0, 1.6))
o2 = self.o.rebound(b)
self.assertEqual(o2.bounds, b)
p = (0.653, -0.234, 0.113)
l1 = list(self.o.by_distance_from_point(p))
l2 = list(o2.by_distance_from_point(p))
self.assertEqual(
l1, l2, "enlarging the bounds shouldn't have changed the points")
def test_restrict(self):
b = ((-1.57, 0.43), (-0.76, 0.83), (-0.37, 1.96))
o2 = self.o.rebound(b)
self.assertEqual(o2.bounds, b)
p = (0.653, -0.234, 0.113)
l1 = [(d, c, v) for (d, c, v) in self.o.by_distance_from_point(p)
if point_in_box(c, b)]
l2 = list(o2.by_distance_from_point(p))
self.assertEqual(
l1, l2,
"playing around with the bounds should restrict the points")
def test_remove_all(self):
p = (-0.432, 0.651, 0.791)
l = list(self.o.by_distance_from_point(p))
n = len(l)
for (i, (_, c, _)) in enumerate(l):
self.o.remove(c)
self.assertEqual(len(self.o), n - i - 1)
with self.assertRaises(KeyError):
self.o.remove(c)
