如何在VS Code中统一pytest工作目录并解决测试发现失效问题

问题背景 #

项目目录结构如下：

project-root
|- .vscode
   - settings.json
|-data-processing
| |-src
| |-tests

在data-processing目录下执行pytest tests时，os.getcwd()返回...project-root/data-processing，但VS Code默认运行pytest时工作目录是project-root，导致路径依赖的测试（比如PySpark相关）出错。之前尝试设置python.testing.cwd为./data-processing/并修改pytestArgs为["tests"]后测试完全失效，且无法通过os.chdir临时改目录。

可行解决方案 #

方案1：用VS Code工作区变量配置settings.json #

直接在.vscode/settings.json中使用${workspaceFolder}变量指定绝对路径，避免相对路径的歧义：

{
    "python.testing.pytestArgs": [
        "tests"
    ],
    "python.testing.unittestEnabled": false,
    "python.testing.pytestEnabled": true,
    "python.testing.cwd": "${workspaceFolder}/data-processing"
}

这个配置会让VS Code在data-processing目录下执行pytest tests，和命令行行为完全一致。注意要确保VS Code打开的工作区是project-root，否则${workspaceFolder}会指向其他路径。

方案2：在data-processing下添加pytest配置文件 #

在data-processing目录下创建pytest.ini文件，内容如下：

[pytest]
testpaths = tests

然后修改.vscode/settings.json的pytestArgs为指向data-processing目录：

{
    "python.testing.pytestArgs": [
        "data-processing"
    ],
    "python.testing.unittestEnabled": false,
    "python.testing.pytestEnabled": true
}

当pytest运行时，会自动读取data-processing下的pytest.ini，并以该目录作为工作目录执行测试，无需额外设置python.testing.cwd。

方案3：用VS Code任务手动指定工作目录 #

如果上述配置仍有问题，可以创建自定义任务来运行pytest：
在.vscode/tasks.json中添加以下内容（如果没有该文件则新建）：

{
    "version": "2.0.0",
    "tasks": [
        {
            "label": "run pytest in data-processing",
            "type": "shell",
            "command": "pytest tests",
            "options": {
                "cwd": "${workspaceFolder}/data-processing"
            },
            "group": {
                "kind": "test",
                "isDefault": true
            },
            "problemMatcher": ["$pytest"]
        }
    ]
}

之后可以通过Ctrl+Shift+B（Windows/Linux）或Cmd+Shift+B（Mac）选择该任务运行，测试会在正确的工作目录下执行，同时VS Code的问题面板也会显示测试结果。

注意事项 #

• 确保VS Code使用的Python解释器和你在命令行中使用的是同一个，避免环境依赖问题。
• 如果使用虚拟环境，要先在VS Code中激活对应的虚拟环境。

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