如何编写验证Python应用安装可行性的单元测试？

刚好碰到过类似的需求，我来帮你把这个安装测试的代码补全！先理下你的场景：你的Python应用目录结构是这样的，想在application/test目录下写一个单元测试，验证应用能不能正常安装，要是安装失败就直接标记测试不通过。

.
├── application
│   ├── __init__.py
│   ├── code
│   │   ├── ...
│   └── test
│       ├── ...
└── setup.py

下面是完善后的完整测试代码，每部分我都加了注释和说明：

import unittest
import os
import subprocess
import sys

class TestInstallation(unittest.TestCase):
    def test_installation_success(self):
        # 拿到setup.py所在的项目根目录：当前测试脚本在application/test，所以往上跳两级
        project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
        
        # 用当前Python环境的pip执行安装，避免不同环境的pip导致的问题
        install_cmd = [sys.executable, "-m", "pip", "install", "."]
        
        try:
            # 执行安装命令，check=True会在命令返回非0状态时抛出异常
            subprocess.run(
                install_cmd,
                cwd=project_root,
                check=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True
            )
            # 走到这一步说明安装成功，直接通过测试
            self.assertTrue(True, "应用安装成功")
        except subprocess.CalledProcessError as e:
            # 捕获安装失败的异常，把错误信息打印出来并标记测试失败
            error_details = f"安装失败，错误输出：\n{e.stderr}"
            self.fail(error_details)

if __name__ == '__main__':
    unittest.main()

关键细节说明： #

• 路径处理：用os.path.dirname(__file__)获取当前测试脚本的目录，再通过../../定位到项目根目录（也就是setup.py所在的位置），转成绝对路径是为了避免相对路径在不同执行场景下出错。
• 环境一致性：用sys.executable调用当前运行测试的Python解释器对应的pip，这样能确保测试用的是你当前的开发环境，不会不小心用到系统其他Python环境的pip，导致测试结果不准。
• 命令执行与错误捕获：subprocess.run的check=True参数会让命令执行失败时直接抛出CalledProcessError，我们捕获这个异常后，用self.fail()把错误信息传递给unittest，这样测试报告会明确显示安装失败的原因。
• 输出捕获：stdout和stderr参数把命令的输出捕获下来，方便在失败时查看具体的错误日志，比如依赖缺失、setup.py语法错误之类的问题。

可选的优化点： #

如果你不想测试后把应用留在环境里，可以在测试方法末尾加个卸载命令（记得替换成你的包名）：

# 卸载应用，替换成setup.py里定义的包名
uninstall_cmd = [sys.executable, "-m", "pip", "uninstall", "-y", "your-package-name"]
subprocess.run(uninstall_cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)

或者用unittest的tearDown方法来统一处理卸载，这样即使测试中途失败，也能尝试清理环境。

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