Python如何测试包含随机数生成的算法_固定随机种子并使用pytest校验

Python如何测试包含随机数生成的算法：固定随机种子并使用pytest校验

测试包含随机数生成的代码，失败的根本原因在于随机结果不可重现。解决这个问题的核心，不在于如何生成随机数，而在于如何让随机过程变得可控。关键在于固定各依赖库的随机种子，并用pytest fixture进行统一管理，同时将断言的重点从具体的输出值，转向校验其统计性质。

为什么测试含 random 的代码总失败？

道理很简单：如果每次运行的结果都不同，那么基于具体值的断言就注定会失败。问题的核心并非“如何生成随机数”，而是“如何让随机变得可重现”。真正的突破口在于控制随机源——我们的目标不是禁用random，而是精确地固定它的起点。

用 random.seed() 固定种子的正确姿势

在测试函数开头直接调用random.seed(42)，这是一个常见的误区。这种做法只能影响Python标准库random模块的全局状态，却无法覆盖numpy.random、secrets或第三方库（比如torch.manual_seed）的随机源。正确的做法是根据实际依赖进行精准干预：

• 纯Python标准库 → 调用 random.seed(42)
• 用了 numpy → 必须额外加上 np.random.seed(42)（注意：新版推荐使用np.random.Generator，但大量遗留代码仍在使用此方式）
• 涉及 torch → 别忘了补上 torch.manual_seed(42)
• 在pytest中，不建议在每个测试函数内部反复调用seed()，更好的做法是使用fixture进行统一管理。

pytest 中用 fixture 封装种子重置逻辑

手动在每个测试用例里编写random.seed()，不仅容易遗漏，也难以维护。利用pytest的fixture功能，可以确保每次测试执行前，随机环境都是干净且一致的：

@pytest.fixture(autouse=True)
def reset_random_seeds():
    random.seed(42)
    np.random.seed(42)
    if 'torch' in sys.modules:
        import torch
        torch.manual_seed(42)

这个名为reset_random_seeds的fixture设置了autouse=True，意味着所有测试函数都会自动应用它。需要留意的是，它并不保证线程安全。如果测试是并行运行的（例如使用了pytest -n auto），则需要考虑改用模块级或会话级fixture，并配合锁机制。

立即学习“Python免费学习笔记（深入）”；

校验输出时别只比对“一个值”

即使固定了随机种子，某些算法（例如采样、蒙特卡洛模拟）也可能因为底层实现的细微差别，返回不同但逻辑上等价的结果。举个例子：random.sample([1,2,3,4], 2)在种子42下可能固定返回[3, 1]，但如果算法内部是先打乱列表再切片，一个等价的实现可能会产生相同概率分布但顺序不同的结果。

• 优先校验统计性质：例如，将算法运行1000次，检查其输出的均值是否落在理论期望值±可接受的容差范围内。
• 对确定性子步骤进行打桩（Mock）：比如，将random.random()函数替换为返回固定序列的模拟对象。
• 避免严格的相等断言：如果顺序无关紧要，不要用assert result == expected_list，可以改用set(result) == set(expected_list)。
• 断言输出空间的约束：如果算法本身允许多种合法的输出形式，那么断言应该覆盖这些输出必须满足的条件，而非某一个具体的值。

最容易被忽略的一点是：即便重置了所有随机种子，如果测试过程中调用了外部服务或读取了未被模拟（mock）的文件，这些非随机的变量依然会导致结果发生漂移——随机性只是众多影响因素中的一个而已。