def test_enqueue(self): q = Queue() self.assertTrue(q.is_empty()) q.enqueue(1) self.assertFalse(q.is_empty()) self.assertEqual(q.peek(), 1) q.enqueue(2) self.assertNotEqual(q.peek(), 2)
def test_is_empty(self): q = Queue() self.assertTrue(q.is_empty()) for i in range(1000): n = random.randint(0, 1000) q.enqueue(n) counter = 0 while (not q.is_empty()): q.dequeue() counter += 1 self.assertEqual(counter, 1000)
def test_dequeue(self): q = Queue() self.assertRaises(IndexError, q.dequeue) for i in range(10): q.enqueue(i) for i in range(9): self.assertEqual(q.peek(), i) q.dequeue() self.assertNotEqual(q.peek(), i) self.assertFalse(q.is_empty()) q.dequeue() self.assertTrue(q.is_empty())