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_is_empty(self): queue = SimpleQueue() assert queue.is_empty() is True queue.add(1) assert queue.is_empty() is False queue.dequeue() assert queue.is_empty() is True
def test_print(self): queue = SimpleQueue() assert str(queue) == '' queue.add(3) queue.add(1) queue.add(2) assert str(queue) == '3 -> 1 -> 2'
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