def test_should_reorganing_max_after_being_removed(self): a = [3, 5, 8, 2, 1, 9] stack = HeapStack() stack.push_all(a) self.assertEquals(9, stack.find(0)) self.assertEquals(9, stack.top()) stack.pop() self.assertEquals(8, stack.find(0)) self.assertEquals(5, stack.find(1))
