如何在VS Code中通过pytest检测当前运行的是单条测试还是多条测试

我之前也碰到过一模一样的需求——在VS Code里用pytest跑测试时，单测要保留调试输出方便排查，批量并行跑的时候就把那些耗时的输出关掉。分享几个亲测有效的方法给你：

这是最准确的方式，因为它直接从pytest的测试收集结果里拿到实际要运行的测试用例数量，不管你是通过VS Code点单测按钮，还是手动输入命令行参数，都能准确判断。

你只需要在项目根目录下新建（或修改）conftest.py，添加这个钩子函数：

def pytest_collection_finish(session):
    # 统计本次运行的测试用例总数
    session.config._total_tests = len(session.items)

然后在你的测试代码或者业务代码里，就可以通过pytest的config对象拿到这个数值：

import pytest

def your_debug_function():
    total_tests = pytest.config._total_tests
    if total_tests == 1:
        # 单条测试运行，保留调试输出
        print("[DEBUG] 详细的调试信息...")
    else:
        # 多条/并行测试，跳过耗时输出
        pass

VS Code运行单条测试时，会给pytest传递带::的命令行参数（比如pytest test_demo.py::TestDemo::test_case），而运行多条测试时，参数通常是文件路径或者更宽泛的范围。我们可以通过解析命令行参数来判断：

import sys

def is_running_single_test():
    # 遍历除了pytest本身之外的命令行参数
    for arg in sys.argv[1:]:
        # 排除掉pytest的选项参数（比如-v、-s这些）
        if not arg.startswith("-") and "::" in arg:
            return True
    return False

# 使用示例
if is_running_single_test():
    enable_debug_output()
else:
    disable_debug_output()

这个方法的好处是不需要修改conftest.py，缺点是如果用户手动输入了特殊的命令行参数（比如指定多个单测节点），可能会误判，但在VS Code的常规使用场景下完全够用。

• 如果你用的是pytest 7.x以上版本，session.items的获取方式是稳定的，但如果是更老的版本，可能需要调整为session.testscollected或者其他属性，建议先打印dir(session)看看可用属性。
• 第一种方法里的_total_tests是自定义的属性名，只要不和pytest本身的属性冲突就行。

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