def test_size(self):
queue = SimpleQueue()
assert queue.size() == 0
queue.push(1)
assert queue.size() == 1
queue.pop()
assert queue.size() == 0
def test_enqueue(self):
queue = SimpleQueue()
tests = [1, '0', SimpleQueue, lambda x: x, {}, [], None]
for test in tests:
queue.add(test)
assert len(tests) == queue.size()
def test_dequeue(self):
queue = SimpleQueue()
with self.assertRaises(SimpleQueueDequeueException):
queue.dequeue()
one = 1
two = 2
queue.add(one)
queue.add(two)
assert queue.size() == 2
assert queue.dequeue() == one
assert queue.dequeue() == two
assert queue.size() == 0
