Queue

from my_array_deque import ArrayDeque
from my_linked_list import LinkedList

class Queue:
    def __init__(self, a_type = None):
        ''' Initalises either a linked list or deque '''
        self.type = a_type
        if self.type == 'array':
            self.queue = ArrayDeque()
        elif self.type == 'linked':
            self.queue = LinkedList()
        else:
            raise TypeError

    def __str__(self):
        return self.queue.__str__()

    def add(self, value):
        return self.queue.push_back(value)

    def remove(self):
        return self.queue.pop_front()

    def get_size(self):
        return self.queue.get_size()