如何用pytest断言学生编写的Python代码与预期一致？

原测试的问题分析 #

你当前的测试代码逻辑完全错误：print1是函数对象，和字符串"Hello world"用is比较永远不成立；而且print()函数本身返回None，所以调用print1(x)的结果也是None，没法通过返回值直接判断。

下面分两种场景给出解决方案，同时补充适合编程教学的替代工具：

场景1：验证功能正确性（推荐，符合计算思维考核） #

如果核心目标是确认学生代码实现了「打印Hello world」的功能，而非严格限制代码写法，推荐捕获标准输出进行验证：

# test_main.py
from io import StringIO
import sys
from main import print1

def test_print1_output():
    # 临时替换标准输出为内存缓冲区
    captured_output = StringIO()
    sys.stdout = captured_output
    
    # 调用学生的函数（参数随便传，原函数未用到x）
    print1("dummy_arg")
    
    # 恢复标准输出
    sys.stdout = sys.__stdout__
    
    # 验证输出内容（strip()处理换行符）
    assert captured_output.getvalue().strip() == "Hello world"

这种方法关注代码的实际效果，符合TDD核心思路——只要功能达标就通过，不会限制学生的实现灵活性。

场景2：严格验证代码语句（适合语法教学） #

如果必须确认学生准确写出了print("Hello world")并作为返回值，可以通过解析函数源代码实现：

# test_main.py
import inspect
import re
from main import print1

def test_print1_code():
    # 获取学生函数的源代码
    source_code = inspect.getsource(print1)
    
    # 用正则匹配，忽略空格、换行等格式差异
    # 匹配return print("Hello world")的各种写法（比如中间加空格）
    pattern = re.compile(r'return\s+print\s*\(\s*"Hello world"\s*\)', re.MULTILINE)
    assert pattern.search(source_code) is not None, "未找到指定的代码语句"

注意：这种方法对代码格式的容忍度可以通过正则调整，比如允许单引号、忽略多余空格等，避免因格式细节误判学生代码。

编程教学替代工具推荐 #

除了自己搭建pytest套件，这些工具更适合课堂教学场景：

• Replit：支持实时代码运行、批量作业管理，内置测试模板，学生提交后自动运行测试
• Codewars：闯关式练习，自带预设测试用例，适合巩固语法和算法思维
• Gradescope：专业的作业批改平台，支持自定义pytest测试用例，批量批改学生代码
• PyBites：基于真实场景的Python练习，侧重实战能力，自带测试框架

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