如何在VS Code中用pytest检测运行单测还是多测？

方案一：借助pytest会话钩子统计测试总数 #

在项目根目录的conftest.py中添加钩子函数，统计本次测试会话要运行的总测试数，之后在代码里判断该数值是否为1即可。

# conftest.py
def pytest_configure(config):
    # 获取本次会话的所有测试项
    config.total_tests = len(config.items)

在需要控制输出的代码中：

import pytest

def should_show_debug_output():
    # 先判断是否处于测试运行状态，再判断是否是单个测试
    if not is_in_test_mode():  # 这里替换成你已有的检测测试状态的逻辑
        return False
    return getattr(pytest.config, 'total_tests', 0) == 1

# 使用示例
if should_show_debug_output():
    print("[DEBUG] 详细调试信息...")
else:
    # 多测并行时屏蔽输出
    pass

方案二：解析pytest命令行参数 #

VS Code运行单个测试时，pytest的命令行会明确指定具体的测试目标（比如test_demo.py::TestCalc::test_add），通过解析命令行参数可以判断是否仅指定了单个测试项。

import sys

def is_single_test_running():
    # 过滤掉pytest的命令行选项（以-开头的参数），只保留测试目标
    test_targets = [arg for arg in sys.argv[1:] if not arg.startswith('-')]
    if len(test_targets) != 1:
        return False
    target = test_targets[0]
    # 两种情况判定为单个测试：一是包含::（指定到具体方法/类），二是单个文件且无通配符
    return '::' in target or ('.py' in target and '*' not in target)

方案三：结合测试节点ID与会话统计 #

通过pytest_runtest_setup钩子将总测试数绑定到当前测试项上，在测试函数中通过request对象获取该值进行判断。

# conftest.py
def pytest_configure(config):
    config.total_tests = len(config.items)

def pytest_runtest_setup(item):
    # 将总测试数绑定到当前测试项
    item.total_tests = item.config.total_tests

在测试函数中使用：

def test_sample(request):
    if request.node.total_tests == 1:
        print("[DEBUG] 单个测试运行，输出调试日志")
    # 测试逻辑代码

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