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_enqueue():
letters = Queue()
letters.enqueue('A')
letters.enqueue('B')
letters.enqueue('C')
expected = 'A'
assert expected == letters.front.value
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 __init__(self):
self.dogs = Queue()
self.cats = Queue()
def test_queue_empty_exception():
queue = Queue()
with pytest.raises(Exception):
assert queue.dequeue()
with pytest.raises(Exception):
assert queue.peek()
def test_init():
queue = Queue()
assert queue.front == None
assert queue.rear == 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_exception_Q():
letters = Queue()
expected = "Queue Is Empty"
assert expected == letters.dequeue()
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()
def test_enqueue_muliple():
letters = Queue()
letters.enqueue('w','x','y','z')
expected = 'w'
assert expected == letters.peek()