def test_popping(self): pq = PriorityStack() for item, pr in INPUT: pq.push(item, pr) l = [] while pq: l.append(pq.pop()) self.assertEquals(l, self.output)
def test_iter(self): pq = PriorityStack() for item, pr in INPUT: pq.push(item, pr) result = [x for x in pq] self.assertEquals(result, self.output)