def test_PriorityQueue_PopAllSmallerThan_LIstIsEmpty(self): priority_queue = PriorityQueue() priority_queue.add("a", 2) priority_queue.add("b", 1) priority_queue.remove("b") l = priority_queue.pop_smaller_than(3) self.assertEquals(list(priority_queue), [])
def test_PriorityQueue_PopSmallerThan_ReturnsThePopedElements(self): priority_queue = PriorityQueue() priority_queue.add("a", 2) priority_queue.add("b", 3) priority_queue.add("c", 1) priority_queue.add("d", 0) priority_queue.remove("c") priority_queue.add("c", 4) priority_queue.remove("b") priority_queue.add("e", 7) priority_queue.add("f", 5) priority_queue.add("g", 6) priority_queue.add("h", 10) l = priority_queue.pop_smaller_than(6) self.assertEquals(l, [('d', 0), ('a', 2), ('c', 4), ('f', 5)]) self.assertEquals(list(priority_queue), [('g', 6), ('e', 7), ('h', 10)])