def test_update(self): Q = PriorityQueue() Q.add_update(1, 2) Q.add_update(2, 1) Q.add_update(1, 0) self.assertEqual(Q.pop(), 1)
def test_pop(self): Q = PriorityQueue() Q.add_update(1, 1) Q.add_update(2, 0) self.assertEqual(Q.pop(), 2) Q.add_update(3, 0.5) self.assertEqual(Q.pop(), 3) self.assertEqual(Q.pop(), 1)
def test_contains(self): Q = PriorityQueue() Q.add_update(1, 0) self.assertTrue(1 in Q) self.assertFalse(2 in Q)