def test_pop_add(self): fifo = FIFOQueue() fifo.add(1) fifo.add(2) fifo.add(3) self.assertEqual(3, fifo.length()) self.assertEqual(1, fifo.pop()) self.assertEqual(2, fifo.pop()) self.assertEqual(3, fifo.pop())
def test_remove(self): fifo = FIFOQueue() fifo.add(1) fifo.add(2) fifo.add(3) self.assertFalse(fifo.remove(5)) self.assertTrue(fifo.remove(2)) self.assertEqual(2, fifo.length()) self.assertEqual(1, fifo.pop()) self.assertEqual(3, fifo.pop())