Exemplo n.º 1
0
 def test_add_add_one(self):
     h = MinMaxHeap()
     self.assertIsNone(h.add(2))
     self.assertEqual(h.size, 1)
     self.assertFalse(h.is_empty())
     self.assertEqual(h.find_max(), 2)
     self.assertEqual(h.find_min(), 2)
Exemplo n.º 2
0
    def test_add_multiple_elements(self):
        a = [randint(-100, 100) for _ in range(100)]
        h = MinMaxHeap()

        for i, elem in enumerate(a):
            self.assertIsNone(h.add(elem))
            self.assertEqual(h.size, i + 1)

        self.assertFalse(h.is_empty())
        self.assertEqual(h.find_max(), max(a))
        self.assertEqual(h.find_min(), min(a))