def test_init_pop(self): pq = PriorityQueue(lambda i: i.__key__(), self.items) self.assertEqual(pq.peek().x, 0) [self.assertEqual(pq.pop().x, i) for i in range(pq.size)] self.assertEqual(pq.size, 0)