def test_queue(self):
q = Queue()
self.assertTrue(q.is_empty())
self.assertRaises(EmptyQueueException, q.peek)
self.assertRaises(EmptyQueueException, q.remove)
q.add(1)
self.assertFalse(q.is_empty())
q.add(2)
q.add(3)
self.assertEqual(1, q.peek())
self.assertEqual(1, q.remove())
self.assertEqual(2, q.peek())
q.remove()
q.remove()
self.assertTrue(q.is_empty())
self.assertRaises(EmptyQueueException, q.remove)
def test_remove(self):
queue = Queue()
queue.add('one')
queue.add('two')
data = queue.remove()
self.assertEqual(data, 'one')
self.assertEqual(queue.first.data, 'two')
self.assertEqual(queue.last.data, 'two')
def test_remove_empty(self):
queue = Queue()
with self.assertRaises(ValueError):
queue.remove()
