class OctreeTests(unittest.TestCase):
def setUp(self):
self._edge = 100.0
self._count = 5
self._res = 5.0
self._root = OctreeNode((0,0,0),
(self._edge, self._edge, self._edge),
res=5.0)
x = np.linspace(-self._edge/2.0, self._edge/2.0, self._count)
c = 0
for i in x:
for j in x:
for k in x:
tmp = IntContainer()
tmp.pos = (i,j,k)
tmp.value = c
self._root.add_object(tmp, tmp.pos)
#print("%i: (%f,%f,%f)" % (c, i, j, k))
c += 1
def test_octree_find_objects_1(self):
correct_values = [62]
subset = self._root.find_objects((0., 0., 0.), 10.)
values = [x.value for x in subset]
self.assertEqual(set(values), set(correct_values))
def test_octree_find_objects_2(self):
correct_values = [31, 32, 33, 36, 37, 38, 41, 42,
43, 56, 57, 58, 61, 62, 63, 66, 67, 68, 81,
82, 83, 86, 87, 88, 91, 92, 93]
subset = self._root.find_objects((0., 0., 0.), 26.)
values = [x.value for x in subset]
self.assertEqual(set(values), set(correct_values))
def setUp(self):
self._edge = 100.0
self._count = 5
self._res = 5.0
self._root = OctreeNode((0, 0, 0),
(self._edge, self._edge, self._edge),
res=5.0)
x = np.linspace(-self._edge / 2.0, self._edge / 2.0, self._count)
c = 0
for i in x:
for j in x:
for k in x:
tmp = IntContainer()
tmp.pos = (i, j, k)
tmp.value = c
self._root.add_object(tmp, tmp.pos)
#print("%i: (%f,%f,%f)" % (c, i, j, k))
c += 1
class OctreeTests(unittest.TestCase):
def setUp(self):
self._edge = 100.0
self._count = 5
self._res = 5.0
self._root = OctreeNode((0, 0, 0),
(self._edge, self._edge, self._edge),
res=5.0)
x = np.linspace(-self._edge / 2.0, self._edge / 2.0, self._count)
c = 0
for i in x:
for j in x:
for k in x:
tmp = IntContainer()
tmp.pos = (i, j, k)
tmp.value = c
self._root.add_object(tmp, tmp.pos)
#print("%i: (%f,%f,%f)" % (c, i, j, k))
c += 1
def test_octree_find_objects_1(self):
correct_values = [62]
subset = self._root.find_objects((0., 0., 0.), 10.)
values = [x.value for x in subset]
self.assertEqual(set(values), set(correct_values))
def test_octree_find_objects_2(self):
correct_values = [
31, 32, 33, 36, 37, 38, 41, 42, 43, 56, 57, 58, 61, 62, 63, 66, 67,
68, 81, 82, 83, 86, 87, 88, 91, 92, 93
]
subset = self._root.find_objects((0., 0., 0.), 26.)
values = [x.value for x in subset]
self.assertEqual(set(values), set(correct_values))
def setUp(self):
self._edge = 100.0
self._count = 5
self._res = 5.0
self._root = OctreeNode((0,0,0),
(self._edge, self._edge, self._edge),
res=5.0)
x = np.linspace(-self._edge/2.0, self._edge/2.0, self._count)
c = 0
for i in x:
for j in x:
for k in x:
tmp = IntContainer()
tmp.pos = (i,j,k)
tmp.value = c
self._root.add_object(tmp, tmp.pos)
#print("%i: (%f,%f,%f)" % (c, i, j, k))
c += 1
