class TestHeap(unittest.TestCase): def setUp(self): self.testHeap = Heap() self.testHeap._data = [100, 84, 71, 60, 23, 12, 29, 1, 37, 4] self.testHeap._heap_size = 10 def tearDown(self): del self.testHeap def test_insert(self): correct = [100, 90, 71, 60, 84, 12, 29, 1, 37, 4, 23] self.testHeap.insert(90) self.assertEqual(self.testHeap._data, correct) self.assertEqual(len(self.testHeap), 11) def test_extract(self): correct = [84, 60, 71, 37, 23, 12, 29, 1, 4] output = self.testHeap.extract() self.assertEqual(self.testHeap._data, correct) self.assertEqual(len(self.testHeap), 9) self.assertEqual(output, 100)
def setUp(self): self.testHeap = Heap() self.testHeap._data = [100, 84, 71, 60, 23, 12, 29, 1, 37, 4] self.testHeap._heap_size = 10