def testInsert(self): u"""测试堆的插入操作""" heap = Heap([4, 6, 5, 7]) heap.insert(3) self.assertEqual(5, heap.size()) self.assertEqual([3, 4, 5, 7, 6], heap._heap) heap.insert(1) self.assertEqual(6, heap.size()) self.assertEqual([1, 4, 3, 7, 6, 5], heap._heap)
def testExtract(self): u"""测试堆的提取最值操作""" heap = Heap([1, 2, 3, 4, 5, 6]) data = heap.extract() self.assertEqual(1, data) self.assertEqual(5, heap.size()) self.assertEqual([2, 4, 3, 6, 5], heap._heap) data = heap.extract() self.assertEqual(2, data) self.assertEqual(4, heap.size()) self.assertEqual([3, 4, 5, 6], heap._heap)
def testSize(self): u"""测试堆的获取尺寸方法""" heap = Heap(range(5)) self.assertEqual(5, heap.size())