def setUp(self): self.Queues = Queues()
class QueueTest(unittest.TestCase): def setUp(self): self.Queues = Queues() def test_dequeue(self): # Dequeue Queues for num in range(1, 6): self.Queues.enqueue(num) for num in range(1, 5): self.assertEquals(self.Queues.dequeue(), num) self.assertEquals(self.Queues.tail.value, 5) # Test last Queue self.assertEquals(dir(self.Queues.tail), dir(self.Queues.head)) self.Queues.dequeue() # Test Empty Queue Exception self.Queues.print_all() with self.assertRaises(IndexError): self.Queues.dequeue() def test_enqueue(self): # Initial Value Test self.Queues.enqueue(1) self.assertEquals(dir(self.Queues.tail), dir(self.Queues.head)) # Test other queues for num in range(2, 6): self.Queues.enqueue(num) # self.Queues.print_all() self.assertEqual(self.Queues.head.value, 1) self.assertEqual(self.Queues.tail.value, num) def test_size(self): # Test incremental size for num in range(1, 6): self.Queues.enqueue(num) self.assertEqual(self.Queues.size(), num) # Test decremental size for num in range(5, 1, -1): self.assertEqual(self.Queues.size(), num) self.Queues.dequeue() self.assertEqual(dir(self.Queues.head), dir(self.Queues.tail)) # Test final empty Queue self.Queues.dequeue() self.assertEqual(self.Queues.size(), 0)