Example #1
0
    def test_peek(self):
        stack = Stack()
        self.assertRaises(Exception, stack.peek)

        stack.push(5)
        stack.push(4)
        self.assertEqual(stack.peek(), 4)
        self.assertEqual(stack.size(), 2)
        stack.pop()
        self.assertEqual(stack.peek(), 5)
        self.assertEqual(stack.size(), 1)