def test_after_single_push_find_max_returns_value(self): heap = MaxHeap() heap.insert(0) assert heap.find_max() == 0
def test_after_two_pushes_find_max_finds_larger_value(self): heap = MaxHeap() heap.insert(0) heap.insert(1) assert heap.find_max() == 1
def test_find_max_of_empty_heap_raises_exception(self): heap = MaxHeap() with raises(MaxHeap.EmptyHeapException): heap.find_max()