def testGetSegmentCounts(self): ssm = SegmentSparseMatrix(2048, 1000) created = ssm.createSegments([42, 42, 43, 44, 45, 43]) ssm.destroySegments([created[2], created[4]]) np.testing.assert_equal(ssm.getSegmentCounts([42, 43, 44, 45, 46]), [2, 1, 1, 0, 0])
def testNoRowLeaks(self): ssm = SegmentSparseMatrix(2048, 1000) created = ssm.createSegments([42, 43, 44, 45, 46]) ssm.destroySegments([created[1], created[2], created[3]]) ssm.createSegments([50, 51, 52, 53]) self.assertEquals(6, ssm.matrix.nRows())