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)
