pytest是否支持参数化导入语句，适配不同测试运行环境？

完全可以通过参数化逻辑实现单测试文件适配多环境导入，不需要维护两份重复的测试代码，以下是两种常用的落地方式：

方案1：pytest自定义命令行参数（灵活度更高） #

通过注册pytest自定义运行参数，执行测试时直接指定环境即可自动切换导入路径：

1. 在项目根目录新建conftest.py文件，注册自定义运行参数：

def pytest_addoption(parser):
    parser.addoption(
        "--env",
        action="store",
        default="local",
        help="指定运行环境：local为本地开发环境，pants为远程Pants构建环境"
    )

2. 修改原有test.py文件，将固定导入替换为动态导入逻辑，原有测试用例代码完全不需要修改：

import importlib
import pytest

# 会话启动时根据参数导入对应模块
def pytest_sessionstart(session):
    global my_code
    run_env = session.config.getoption("--env")
    if run_env == "local":
        my_code = importlib.import_module(".my_code", package=__name__)
    elif run_env == "pants":
        my_code = importlib.import_module("long.path.to.my_project.my_code")

# 原有测试逻辑保持不变
def test_function():
    # 直接调用my_code的方法即可
    assert my_code.your_func() == expected_result

3. 执行方式：

• 本地运行直接执行原有命令，默认使用本地导入路径：
  python -m pytest
• 远程Pants环境运行时指定参数即可自动切换导入路径：
  python -m pytest --env=pants

方案2：环境变量控制（无需额外配置文件） #

如果不想新增conftest.py，也可以通过系统环境变量控制导入逻辑：

1. 直接修改test.py的导入部分：

import os

run_env = os.getenv("TEST_ENV", "local")
if run_env == "local":
    from . import my_code
else:
    import long.path.to.my_project.my_code as my_code

# 原有测试逻辑完全不变
def test_function():
    assert my_code.your_func() == expected_result

2. 执行方式：

• 本地运行直接执行原有命令即可：
  python -m pytest
• 远程Pants环境运行前先设置环境变量即可：

# Linux/macOS 环境
export TEST_ENV=pants && python -m pytest
# Windows 环境
set TEST_ENV=pants && python -m pytest

两种方案都可以实现100%复用测试逻辑，不需要维护重复的测试文件，后续如果新增其他环境，只需要增加对应导入分支即可。

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