Esempio n. 1
0
    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), [])
Esempio n. 2
0
    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)])