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)