def test_enqueue(): letters = Queue() letters.enqueue('A') letters.enqueue('B') letters.enqueue('C') expected = 'A' assert expected == letters.front.value
def test_is_empty(): queue = Queue() queue.enqueue(7) queue.enqueue(3) queue.dequeue() queue.dequeue() assert queue.isEmpty() == True
def test_peek_Q(): letters = Queue() letters.enqueue('A') letters.enqueue('B') letters.enqueue('C') expected = 'A' actual = letters.peek() assert expected == actual
def test_empty_Q(): letters = Queue() letters.enqueue('A') letters.enqueue('B') letters.enqueue('C') letters.dequeue() letters.dequeue() letters.dequeue() expected = True assert expected == letters.is_empty()
class AnimalShelter: def __init__(self): self.dogs = Queue() self.cats = Queue() def enqueue(self, obj): try: if obj.type == 'cat': self.cats.enqueue(obj) elif obj.type == 'dog': self.dogs.enqueue(obj) except AttributeError as e: return 'You Should Add Cat or Dog Only' def dequeue(self, pref): try: if pref == 'cat': cat = self.cats.dequeue() return cat.name elif pref == 'dog': dog = self.dogs.dequeue() return dog.name except AttributeError as e: return None
def test_queue_peek(): queue = Queue() queue.enqueue(7) queue.enqueue(3) assert queue.peek() == 7
def test_dequeue(): queue = Queue() queue.enqueue(7) queue.enqueue(3) queue.dequeue() assert queue.front.value == 3
def test_enqueue_multiple(): queue = Queue() queue.enqueue(7) queue.enqueue(3) assert queue.rear.value == 3 assert queue.front.value == 7
def test_enqueue_one(): queue = Queue() queue.enqueue(7) assert queue.rear.value == 7
def test_enqueue_muliple(): letters = Queue() letters.enqueue('w','x','y','z') expected = 'w' assert expected == letters.peek()