def setUp(self):
self.s = SqStack(5)
class TestSqStack(unittest.TestCase):
def setUp(self):
self.s = SqStack(5)
def tearDown(self):
print 'test has been done'
def test_get_top(self):
self.assertRaises(ValueError, self.s.get_top)
self.s.data[0] = 111
self.s.top += 1
self.assertEquals(111, self.s.get_top())
def test_push(self):
self.s.push(111)
self.assertEquals(1, self.s.get_length())
self.assertEquals(111, self.s.get_top())
self.s.push(222)
self.s.push(333)
self.s.push(444)
self.s.push(555)
self.assertRaises(ValueError, self.s.push, 666)
def test_pop(self):
self.assertRaises(ValueError, self.s.pop)
self.s.push(111)
self.assertEquals(111, self.s.pop())
self.s.push(222)
self.s.push(333)
self.s.push(444)
self.s.push(555)
self.assertEquals(555, self.s.pop())
self.assertEquals(444, self.s.pop())
self.assertEquals(333, self.s.pop())
self.assertEquals(222, self.s.pop())
self.assertRaises(ValueError, self.s.pop)
