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()