如何将文件内容喂给Python的input()并捕获print()输出做自动化测试

问题1：能否在保留主程序input()逻辑的前提下，将.in文件内容喂给input() #

完全可以，核心思路是重定向标准输入流sys.stdin，无需修改主程序的任何输入逻辑。
你可以先读取.in文件的全部内容存入内存缓冲区，再将sys.stdin替换为缓冲区对象，后续主程序调用input()时会自动从缓冲区读取内容，和从终端读取的行为完全一致。示例代码如下：

import sys
from io import StringIO
import your_main_module # 导入你的主程序模块

# 读取测试用例输入
with open("case01.in", "r", encoding="utf-8") as f:
    test_input = f.read()

# 重定向标准输入到内存缓冲区
origin_stdin = sys.stdin
sys.stdin = StringIO(test_input)

# 运行主程序
your_main_module.main()

# 测试完成后恢复标准输入
sys.stdin = origin_stdin

问题2：能否捕获主程序print()的输出并存储到文件 #

可以，和输入重定向逻辑类似，重定向标准输出流sys.stdout到内存缓冲区即可捕获所有print()的输出，运行结束后将缓冲区内容写入文件或者直接和预期的.out文件对比即可。示例代码如下：

import sys
from io import StringIO
import your_main_module

with open("case01.in", "r", encoding="utf-8") as f:
    test_input = f.read()

origin_stdin = sys.stdin
origin_stdout = sys.stdout

# 初始化输入输出缓冲区
sys.stdin = StringIO(test_input)
output_buf = StringIO()
sys.stdout = output_buf

# 运行主程序
your_main_module.main()

# 恢复标准流
sys.stdin = origin_stdin
sys.stdout = origin_stdout

# 获取实际输出并存储
actual_output = output_buf.getvalue()
with open("case01_actual.out", "w", encoding="utf-8") as f:
    f.write(actual_output)

# 也可以直接和预期输出对比
with open("case01.out", "r", encoding="utf-8") as f:
    expect_output = f.read()
if actual_output.strip() == expect_output.strip():
    print("测试用例通过")
else:
    print("测试用例不通过")

如果你不想通过导入模块的方式运行主程序，也可以用subprocess调用主程序文件，天然隔离输入输出：

import subprocess
with open("case01.in", "r", encoding="utf-8") as f:
    test_input = f.read()
res = subprocess.run(
    ["python", "your_main.py"],
    input=test_input,
    capture_output=True,
    text=True,
    encoding="utf-8"
)
actual_output = res.stdout

问题3：多行输入循环读写场景的注意事项 #

首先你不用担心input()读取的内容被误计入输出：input()是从标准输入流读数据，print()是往标准输出流写数据，两个流完全独立，只要你没有在主程序里主动把读取到的输入内容print出来，输入内容绝对不会出现在输出结果里。
需要注意的问题如下：

• 输入文件的换行符要保留，不要手动去掉每行末尾的\n，否则input()可能会把多行内容拼成一行读取，导致逻辑错误
• 对比输出的时候建议先去掉两端的空白字符再对比，避免因为末尾多余空行、行尾空格等格式问题导致测试误判
• 如果主程序有处理异常输入的逻辑，要确保测试用例的输入格式和要求完全一致，避免缓冲区抛出EOFError导致程序异常退出
• 如果用subprocess方式调用主程序，不要忘记设置text=True参数，否则输出会是字节类型，需要额外解码

内容的提问来源于stack exchange，提问作者Mark J