mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-15 22:08:02 +00:00
@@ -1,43 +1,43 @@
|
||||
from nose.tools import assert_equal
|
||||
import unittest
|
||||
|
||||
|
||||
class TestMinHeap(object):
|
||||
class TestMinHeap(unittest.TestCase):
|
||||
|
||||
def test_min_heap(self):
|
||||
heap = MinHeap()
|
||||
assert_equal(heap.peek_min(), None)
|
||||
assert_equal(heap.extract_min(), None)
|
||||
self.assertEqual(heap.peek_min(), None)
|
||||
self.assertEqual(heap.extract_min(), None)
|
||||
heap.insert(20)
|
||||
assert_equal(heap.array[0], 20)
|
||||
self.assertEqual(heap.array[0], 20)
|
||||
heap.insert(5)
|
||||
assert_equal(heap.array[0], 5)
|
||||
assert_equal(heap.array[1], 20)
|
||||
self.assertEqual(heap.array[0], 5)
|
||||
self.assertEqual(heap.array[1], 20)
|
||||
heap.insert(15)
|
||||
assert_equal(heap.array[0], 5)
|
||||
assert_equal(heap.array[1], 20)
|
||||
assert_equal(heap.array[2], 15)
|
||||
self.assertEqual(heap.array[0], 5)
|
||||
self.assertEqual(heap.array[1], 20)
|
||||
self.assertEqual(heap.array[2], 15)
|
||||
heap.insert(22)
|
||||
assert_equal(heap.array[0], 5)
|
||||
assert_equal(heap.array[1], 20)
|
||||
assert_equal(heap.array[2], 15)
|
||||
assert_equal(heap.array[3], 22)
|
||||
self.assertEqual(heap.array[0], 5)
|
||||
self.assertEqual(heap.array[1], 20)
|
||||
self.assertEqual(heap.array[2], 15)
|
||||
self.assertEqual(heap.array[3], 22)
|
||||
heap.insert(40)
|
||||
assert_equal(heap.array[0], 5)
|
||||
assert_equal(heap.array[1], 20)
|
||||
assert_equal(heap.array[2], 15)
|
||||
assert_equal(heap.array[3], 22)
|
||||
assert_equal(heap.array[4], 40)
|
||||
self.assertEqual(heap.array[0], 5)
|
||||
self.assertEqual(heap.array[1], 20)
|
||||
self.assertEqual(heap.array[2], 15)
|
||||
self.assertEqual(heap.array[3], 22)
|
||||
self.assertEqual(heap.array[4], 40)
|
||||
heap.insert(3)
|
||||
assert_equal(heap.array[0], 3)
|
||||
assert_equal(heap.array[1], 20)
|
||||
assert_equal(heap.array[2], 5)
|
||||
assert_equal(heap.array[3], 22)
|
||||
assert_equal(heap.array[4], 40)
|
||||
assert_equal(heap.array[5], 15)
|
||||
self.assertEqual(heap.array[0], 3)
|
||||
self.assertEqual(heap.array[1], 20)
|
||||
self.assertEqual(heap.array[2], 5)
|
||||
self.assertEqual(heap.array[3], 22)
|
||||
self.assertEqual(heap.array[4], 40)
|
||||
self.assertEqual(heap.array[5], 15)
|
||||
mins = []
|
||||
while heap:
|
||||
mins.append(heap.extract_min())
|
||||
assert_equal(mins, [3, 5, 15, 20, 22, 40])
|
||||
self.assertEqual(mins, [3, 5, 15, 20, 22, 40])
|
||||
print('Success: test_min_heap')
|
||||
|
||||
|
||||
@@ -47,4 +47,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user