Exemplo n.º 1
0
 def setUp(self):
     self.s = LinkStack()
Exemplo n.º 2
0
class TestLinkStack(unittest.TestCase):

    def setUp(self):
        self.s = LinkStack()

    def tearDown(self):
        print 'test has been down'

    def test_is_empty(self):
        self.assertTrue(self.s.is_empty())

    def test_push(self):

        self.s.push(111)
        self.assertEquals(1, self.s.get_length())

        self.s.push(222)
        self.assertEquals(222, self.s.get_top())

        self.s.push(333)
        self.assertEquals(333, self.s.get_top())

    def test_pop(self):
        self.assertRaises(ValueError, self.s.pop)

        self.s.push(111)
        self.assertEquals(1, self.s.get_length())

        self.assertEquals(111, self.s.pop())
        self.s.push(222)
        self.s.push(333)
        self.assertEquals(2, self.s.get_length())
        self.assertEquals(333, self.s.pop())