如何在变量中捕获单元测试的stdout/stderr？

import unittest
import io
import sys

class TestCaptureOutput(unittest.TestCase):
    def test_stdout(self):
        # 保存原始的stdout
        original_stdout = sys.stdout
        try:
            # 创建一个StringIO对象来捕获输出
            captured_output = io.StringIO()
            sys.stdout = captured_output
            
            # 调用要测试的函数
            print("Hello, World!")
            
            # 获取捕获的输出
            output = captured_output.getvalue()
            
            # 断言输出是否符合预期
            self.assertEqual(output.strip(), "Hello, World!")
        finally:
            # 恢复原始的stdout
            sys.stdout = original_stdout

    def test_stderr(self):
        # 保存原始的stderr
        original_stderr = sys.stderr
        try:
            # 创建一个StringIO对象来捕获错误输出
            captured_error = io.StringIO()
            sys.stderr = captured_error
            
            # 调用会触发错误的函数
            raise ValueError("An error occurred!")
            
            # 获取捕获的错误输出
            error_output = captured_error.getvalue()
            
            # 断言错误输出是否符合预期
            self.assertIn("An error occurred!", error_output)
        except ValueError:
            pass  # 预期的异常
        finally:
            # 恢复原始的stderr
            sys.stderr = original_stderr

if __name__ == '__main__':
    unittest.main()