def test_ArrayQueue(self): queue = ArrayQueue() queue.enqueue(1) queue.enqueue(2) queue.enqueue(3) # test __iter__() it = iter(queue) self.assertEqual(1, next(it)) self.assertEqual(2, next(it)) self.assertEqual(3, next(it)) self.assertRaises(StopIteration, next, it) # test __len__() self.assertEqual(3, len(queue)) # test is_empty() self.assertFalse(queue.is_empty()) # test peek() self.assertEqual(1, queue.peek()) # test dequeue() self.assertEqual(1, queue.dequeue()) self.assertEqual(2, queue.dequeue()) self.assertEqual(3, queue.dequeue()) self.assertTrue(queue.is_empty())