def setUp(self):
self.l = LinkQueue()
class TestLinkQueue(unittest.TestCase):
def setUp(self):
self.l = LinkQueue()
def tearDown(self):
print 'test has been done'
def test_append(self):
self.l.append(111)
self.assertEquals(1, self.l.get_length())
self.l.append(222)
self.assertEquals(111, self.l.front.data)
self.assertEquals(222, self.l.rear.data)
def test_shift(self):
self.assertRaises(ValueError, self.l.shift)
self.l.append(111)
self.assertEquals(111, self.l.shift())
self.assertRaises(ValueError, self.l.shift)
self.l.append(111)
self.l.append(222)
self.l.append(333)
self.assertEquals(111, self.l.shift())
self.assertEquals(222, self.l.shift())
self.assertEquals(333, self.l.shift())
self.assertEquals(0, self.l.get_length())
