如何在PyInstaller打包Python项目前运行单元测试？

PyInstaller的.spec文件本质就是可执行的Python脚本，所以你完全可以在打包流程启动前插入单元测试的执行逻辑——测试通过才继续打包，失败就直接终止，完美避免打包出有问题的程序。

具体实现步骤 #

你只需要把单元测试的代码放在spec文件的最顶部（Analysis对象定义之前）就行，下面给你两种常见测试框架的示例：

1. 用unittest框架的情况 #

假设你的测试文件都放在tests目录下，命名以test_开头，在spec文件里添加这段代码：

# -*- mode: python -*-
block_cipher = None
import os
import sys
import unittest

# ---------------------- 单元测试逻辑 start ----------------------
# 自动发现所有测试用例
test_loader = unittest.TestLoader()
test_suite = test_loader.discover('tests', pattern='test_*.py')

# 执行测试
test_runner = unittest.TextTestRunner(verbosity=2)
test_result = test_runner.run(test_suite)

# 测试失败则终止打包
if not test_result.wasSuccessful():
    print("❌ 单元测试失败，终止打包流程")
    sys.exit(1)
# ---------------------- 单元测试逻辑 end ----------------------

a = Analysis(['src/hub/Driver.py'], pathex=['src'], binaries=[], datas=[], hiddenimports=[], hookspath=[], runtime_hooks=[], excludes=[], win_no_prefer_redirects=False, win_private_assemblies=False...)

2. 用pytest框架的情况 #

如果你的项目用pytest管理测试，代码可以更简洁：

# -*- mode: python -*-
block_cipher = None
import os
import sys
import pytest

# ---------------------- 单元测试逻辑 start ----------------------
# 运行pytest，指定测试目录（可根据需要调整参数）
exit_code = pytest.main(['tests', '-v'])

# pytest的退出码非0表示测试失败
if exit_code != 0:
    print("❌ 单元测试失败，终止打包流程")
    sys.exit(exit_code)
# ---------------------- 单元测试逻辑 end ----------------------

a = Analysis(['src/hub/Driver.py'], pathex=['src'], binaries=[], datas=[], hiddenimports=[], hookspath=[], runtime_hooks=[], excludes=[], win_no_prefer_redirects=False, win_private_assemblies=False...)

注意事项 #

• 确保测试代码的路径正确：比如tests目录是相对于spec文件的位置，要是路径不对，测试会找不到用例。
• 打包环境要和测试环境一致：比如依赖库的版本、Python版本，避免因为环境差异导致测试通过但打包出的程序有问题。
• 可以调整测试的 verbosity 参数，让输出更详细或简洁，方便排查问题。

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