def test_peak(self): queue = Queue() queue.enqueue(1) self.assertEqual(queue.peek(), 1) queue.enqueue(2) self.assertEqual(queue.peek(), 1) queue.enqueue(13) self.assertEqual(queue.peek(), 1) queue.dequeue() self.assertEqual(queue.peek(), 2)
def test_queue(self): q = Queue(1) q.enqueue(2) q.enqueue(3) assert q.peek() == 1 assert q.dequeue() == 1 q.enqueue(4) assert q.dequeue() == 2 assert q.dequeue() == 3 assert q.dequeue() == 4 q.enqueue(5) assert q.peek() == 5