class StackTest(unittest.TestCase):
def setUp(self):
self.Stacks = Stacks()
def test_push(self):
for num in range(1, 6):
self.Stacks.push(num)
self.assertEqual(self.Stacks.top.value, num)
def test_pop(self):
#Test Exception
with self.assertRaises(IndexError):
self.Stacks.pop()
#Test Top stack value
for num in range(1, 6):
self.Stacks.push(num)
for num in range(5, 1, -1):
self.assertEquals(self.Stacks.pop(), num)
def setUp(self):
self.Stacks = Stacks()
