def test_add_multiple_elements(self):
a = [randint(-100, 100) for _ in range(100)]
h = MaxHeap()
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))
def test_add_add_one(self):
h = MaxHeap()
self.assertIsNone(h.add(2))
self.assertEqual(h.size, 1)
self.assertFalse(h.is_empty())
self.assertEqual(h.find_max(), 2)