def test_isInSet(self):
cells = [[0, 0, 0] + p
for p in [[1, 2, 4, 5, 8, 9], [2, 3, 4, 5, 8, 9],
[2, 3, 5, 6, 8, 9], [2, 3, 4, 5, 6, 7]]]
kdt = KdTree(cells=cells, metadata={METADATA.statedimension: 3})
for inSetPoint in [[1, 4, 8], [2.5, 4.5, 9], [2.5, 4.5, 8],
[2.5, 4.5, 6]]:
self.assertTrue(kdt.isInSet(inSetPoint), inSetPoint)
for outSetPoint in [[3.1, 4, 8], [1, 4, 9.2], [2.5, 4.5, 7.5],
[2.5, 4.5, 3]]:
self.assertFalse(kdt.isInSet(outSetPoint), outSetPoint)
def test_toBarGridKernel(self):
cells = [[0, 0, 0] + p for p in [[1, 2, 4, 5, 8, 9], [
2, 3, 4, 5, 8, 9
], [2, 3, 5, 6, 8, 9], [2, 3, 4, 5, 7, 8], [2, 3, 5, 6, 4, 5]]]
kdt = KdTree(cells=cells, metadata={METADATA.statedimension: 3})
bgk = kdt.toBarGridKernel(newOriginCoords=[1, 4, 4],
newOppositeCoords=[3, 6, 9],
intervalNumberperaxis=[2, 2, 5])
self.assertEqual(len(bgk.bars), 12)
bgk = kdt.toBarGridKernel(newOriginCoords=[1.5, 4.5, 4.5],
newOppositeCoords=[2.5, 5.5, 8.5],
intervalNumberperaxis=[1, 1, 4])
self.assertEqual(len(bgk.bars), 4)
def test_getMaxBounds(self):
cells = [[0, 0, 0] + p
for p in [[1, 2, 4, 5, 8, 9], [2, 3, 4, 5, 8, 9],
[2, 3, 5, 6, 8, 9], [2, 3, 4, 5, 6, 7]]]
kdt = KdTree(cells=cells, metadata={METADATA.statedimension: 3})
self.assertTrue(all(kdt.getMaxBounds() == [3., 6., 9.]))
