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