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