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()