コード例 #1
0
ファイル: testKdTree.py プロジェクト: smurail/vino
 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)
コード例 #2
0
ファイル: testKdTree.py プロジェクト: smurail/vino
 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)
コード例 #3
0
ファイル: testKdTree.py プロジェクト: smurail/vino
 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.]))