def test_findmax(self):
"""
Test for Finding Max Value
"""
heap = MaxHeap(10)
for i in range(10):
heap.insert(i)
self.assertEqual(heap.find_max(), 9)
def test_build(self):
"""
Test Build Function with Basic List
"""
heap = MaxHeap()
testlist = [0, 1, 2, 3, 4, 5, 6, 7, 8]
self.assertTrue(heap.build_heap(testlist))
self.assertEqual(heap.heap_contents(), [8, 7, 6, 3, 4, 5, 2, 1, 0])
def test_delete(self):
"""
Test Delete Function
"""
heap = MaxHeap(10)
for i in range(9):
heap.insert(i)
self.assertEqual(heap.del_max(), 8)
self.assertEqual(heap.heap_contents(), [7, 6, 5, 3, 2, 1, 4, 0])
def test_sorts(self):
"""
Testing Heap Sort Function
"""
heap = MaxHeap()
testlist = [10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
templist = testlist[:]
templist.sort()
self.assertEqual(heap.heap_sort_increase(testlist), templist)
def test_insert(self):
"""
Test for Insert Function
"""
heap = MaxHeap(10)
self.assertTrue(heap.is_empty())
self.assertFalse(heap.is_full())
for i in range(10):
heap.insert(i)
self.assertTrue(heap.is_full())
self.assertFalse(heap.is_empty())
def test_get(self):
"""
Test Get Functions
"""
heap = MaxHeap(10)
for i in range(9):
heap.insert(i)
self.assertEqual(heap.get_heap_cap(), 10)
self.assertEqual(heap.get_heap_size(), 9)
self.assertFalse(heap.is_empty())
self.assertFalse(heap.is_full())
self.assertEqual(heap.heap_contents(), [8, 7, 5, 6, 2, 1, 4, 0, 3])
def test_dupli(self):
"""
Test Build Function with Duplicates
"""
heap = MaxHeap()
testlist = [24, 12, 34, 45, 67, 21, 21, 21, 3, 4, 100]
self.assertTrue(heap.build_heap(testlist))
self.assertEqual(heap.heap_contents(),
[100, 67, 34, 45, 24, 21, 21, 21, 3, 4, 12])
heap = MaxHeap(5)
testlist = [0, 1, 2, 3, 6, 7, 8, 12, 4, 15, 76]
self.assertFalse(heap.build_heap(testlist))
heap = MaxHeap()
testlist = [
88, 23, 1003, 666, 5027, 303, 22, 108, 99, 666, 88, 666, 303, 1003,
-2
]
self.assertTrue(heap.build_heap(testlist))
self.assertEqual(heap.heap_contents(), [
5027, 666, 1003, 666, 88, 666, 1003, 108, 99, 23, 88, 303, 303, 22,
-2
])
def test_insert2(self):
"""
Test for Insert Function 2
"""
heap = MaxHeap()
heap.insert(15)
heap.insert(10)
heap.insert(12)
heap.insert(5)
heap.insert(6)
heap.insert(7)
heap.insert(9)
self.assertEqual(heap.heap_contents(), [15, 10, 12, 5, 6, 7, 9])
def test_delete2(self):
"""
Test Delete Function 2
"""
heap = MaxHeap()
heap.insert(103)
heap.insert(98)
heap.insert(77)
heap.insert(28)
heap.insert(34)
heap.insert(7)
heap.insert(50)
heap.insert(2)
self.assertEqual(heap.heap_contents(), [103, 98, 77, 28, 34, 7, 50, 2])
self.assertEqual(heap.del_max(), 103)
self.assertEqual(heap.heap_contents(), [98, 34, 77, 28, 2, 7, 50])