document.write('
Data hosted with ♥ by Pastebin.com - Download Raw - See Original
  1. from heap import Heap
  2. import unittest
  3.  
  4. class TestHeap(unittest.TestCase):
  5.     def test_init(self):
  6.         h = Heap(3)
  7.         self.assertEqual(h.get(), [0])
  8.         self.assertEqual(h.getCurrentSize(), 0)
  9.         self.assertEqual(h.getAllowedSize(), 3)
  10.    
  11.     def test_buildHeap(self):
  12.         h = Heap(10)
  13.         l = [1,2,3,4,5,6,7,8]
  14.         h.buildHeap(l)
  15.         self.assertEqual(h.get(), [0] + l)
  16.        
  17.     def test_insert(self):
  18.         h = Heap(3)
  19.         h.insert(5)
  20.         h.insert(10)
  21.         h.insert(12)
  22.         self.assertEqual(h.get(), [0,5,10,12])
  23.         h.insert(14)
  24.         self.assertEqual(h.get(), [0,10,14,12])
  25.    
  26.     def test_delMin(self):
  27.         h = Heap(4)
  28.         l = [1,2,3,4]
  29.         h.buildHeap(l)
  30.         self.assertEqual(h.delMin(), 1)
  31.         self.assertEqual(h.get(), [0,2,4,3])
  32.         h2 = Heap(4)
  33.         h2.insert(5)
  34.         self.assertEqual(h2.delMin(), 5)
  35.         self.assertEqual(h2.get(), [0])
  36.    
  37. if __name__ == "__main__":
  38.     unittest.main()
');