class TestPriorityQueue(unittest.TestCase): def setUp(self): self.testQueue = PriorityQueue() def tearDown(self): del self.testQueue def test_isEmpty(self): self.assertTrue(self.testQueue.isEmpty()) self.assertEqual(len(self.testQueue), 0) def test_enqueue(self): self.testQueue.enqueue('purple', 5) self.assertFalse(self.testQueue.isEmpty()) self.assertEqual(len(self.testQueue), 1) self.testQueue.enqueue('black', 1) self.assertEqual(len(self.testQueue), 2) def test_dequeue(self): self.testQueue.enqueue('purple', 5) self.testQueue.enqueue('black', 1) self.testQueue.enqueue("white", 0) self.testQueue.enqueue("green", 1) self.testQueue.enqueue("yellow", 5) node1 = self.testQueue.dequeue() self.assertEqual(node1, 'purple') self.assertEqual(len(self.testQueue), 4) node2 = self.testQueue.dequeue() self.assertEqual(node2, 'yellow') self.assertEqual(len(self.testQueue), 3) node3 = self.testQueue.dequeue() self.assertEqual(node3, 'black') self.assertEqual(len(self.testQueue), 2)
def setUp(self): self.testQueue = PriorityQueue()