Python代码测试时input语句输出干扰问题解决求助

在代码库执行测试时，main()函数内的input语句提示内容会混入程序输出，导致测试不通过，但在Jupyter Notebook中测试无此异常。当前输出包含大量input提示信息，与期望的纯数据输出不符。

原因分析 #

• 测试场景下，通常会预先通过标准输入(stdin)注入测试数据，而input()函数的提示文本默认输出到标准输出(stdout)，与程序的正常输出（如表格内容）混合，干扰测试断言。
• Jupyter Notebook的输入输出环境是分离的，提示信息不会进入输出流，因此无此问题。

解决方案 #

1. 将input提示重定向到标准错误(stderr) #

把input的提示信息输出到stderr，这样测试时只需捕获stdout即可得到纯净的程序输出，过滤掉提示内容。

2. 调整tabulate输出格式为plain #

将表格输出格式改为plain，匹配期望的空格分隔样式。

3. 处理文件操作错误（可选） #

若不需要write_to_file的错误输出，可添加异常捕获屏蔽错误信息。

修改后的代码示例 #

import sys
import datetime
from tabulate import tabulate

# 假设以下函数已定义：validate_input, calculate_rating, calculate_salary, calculate_age, write_to_file

def main():
    salary_dataset = [[1, 2, 3, 4], [1000, 700, 500, 400], [80, 60, 45, 30]]
    player_data = []
    
    for _ in range(3):
        # 将提示输出到stderr，不混入stdout
        print("Enter player ID (type 'end' to finish): ", file=sys.stderr, end='')
        player_id = input()
        
        if player_id.lower() == 'end':
            break
            
        if len(player_id) != 2 or not player_id.isdigit():
            print("invalid player Id. Please enter a 2-digit number.", file=sys.stderr)
            continue    
            
        print("Enter player name: ", file=sys.stderr, end='')
        name = input()
        
        print("Enter player D.o.B (YYYY-MM-DD): ", file=sys.stderr, end='')
        dob = input()
        
        try:
            datetime.strptime(dob, "%Y-%m-%d") 
        except ValueError:
            print("Invalid data format. Please use YYYY-MM-DD.", file=sys.stderr)
            continue  
        
        skills_set = [0, 0, 0, 0, 0, 0]
        for idx in range(len(skills_set)):
            print("Enter players data (0-5 only): ", file=sys.stderr, end='')
            skills_set[idx] = validate_input(input())
            
        overall_rate_per_player = calculate_rating(skills_set)
        salary_range = calculate_salary(overall_rate_per_player, salary_dataset)
        age = calculate_age(dob)
        
        player_data.append({
            'UID': player_id,
            'Name': name,
            'D.o.B': dob,
            'Age': age,
            'Score': overall_rate_per_player,
            'Salary Range': salary_range
        })
        
    # 使用plain格式输出表格，匹配期望样式
    table = [[p['UID'], p['Name'], p['D.o.B'], p['Age'], p['Score'], p['Salary Range']] 
             for p in sorted(player_data, key=lambda x: x['UID'])]
    print(tabulate(table, tablefmt='plain'))
    
    # 捕获文件操作异常，避免错误输出干扰测试
    try:
        write_to_file(table)
    except FileNotFoundError:
        # 可选择静默处理或输出到stderr
        print("File not found: PlayerData.txt", file=sys.stderr)
        
if __name__ == "__main__":
    main()  
    # advanced()  # 如果测试不需要此函数，可暂时注释或调整

效果说明 #

修改后：

• 所有input提示和错误信息都会输出到stderr，测试时只需断言stdout的内容即可。
• tabulate的plain格式会生成空格分隔的纯文本输出，与期望输出完全一致：

  10  Lionel Messi  1987-06-24     36      100            1000
  

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