def test_first(self): self.assertEqual(self.num_queue.first(), 1) self.assertEqual(self.letter_queue.first(), "a") empty_queue = Queue() with self.assertRaises(LookupError): empty_queue.first()
class TestQueue(unittest.TestCase): def setUp(self): self.num_queue = Queue() self.num_queue.enqueue(3) self.num_queue.enqueue(2) self.num_queue.enqueue(1) self.letter_queue = Queue() self.letter_queue.enqueue("b") self.letter_queue.enqueue("a") def test_enqueue(self): empty_queue = Queue() empty_queue.enqueue(4) self.assertEqual(len(empty_queue), 1) def test_len(self): self.assertEqual(len(self.num_queue), 3) self.assertEqual(len(self.letter_queue), 2) def test_repr(self): self.assertEqual(repr(self.num_queue), "Queue(1, 2, 3)") self.assertEqual(repr(self.letter_queue), "Queue(a, b)") def test_dequeue(self): first = self.num_queue.dequeue() self.assertEqual(len(self.num_queue), 2) self.assertEqual(first, 1) second = self.num_queue.dequeue() self.assertEqual(second, 2) self.assertEqual(len(self.num_queue), 1) empty_queue = Queue() with self.assertRaises(LookupError): empty_queue.dequeue() def test_first(self): self.assertEqual(self.num_queue.first(), 1) self.assertEqual(self.letter_queue.first(), "a") empty_queue = Queue() with self.assertRaises(LookupError): empty_queue.first()