def test_rotate_queue(self):
queue_elems = ''
queue = SimpleQueue()
queue.enqueue(0)
queue.enqueue(1)
queue.enqueue(2)
queue.enqueue(3)
for i in range(queue.size()):
queue_elems += str(queue.dequeue())
assert queue_elems == '0123'
queue_elems = ''
queue.enqueue(0)
queue.enqueue(1)
queue.enqueue(2)
queue.enqueue(3)
rotate_queue(queue, 3)
for el in range(queue.size()):
queue_elems += str(queue.dequeue())
assert queue_elems == '3012'
def test_size(self):
queue = SimpleQueue()
assert queue.size() == 0
queue.enqueue(0)
queue.enqueue(1)
assert queue.size() == 2
def rotate_queue(queue: SimpleQueue, n):
if queue.size() > 0:
for i in range(n):
queue.enqueue(queue.dequeue())
