如何让pytest coverage仅标记测试函数直接调用的代码为已覆盖？

一、自定义Coverage的追踪逻辑 #

Coverage默认会记录所有执行过的代码行，你需要修改它的追踪规则，只保留测试函数直接调用的代码。具体可以这么做：

• 重写Coverage的trace函数：在追踪代码执行时，通过inspect.stack()获取调用栈，检查当前代码行的直接调用者是不是测试函数（比如函数名以test_开头）。只有当直接调用者是测试函数时，才把该行标记为已覆盖。
• 举个简化的实现例子：

  import inspect
  from coverage import Coverage
  
  class DirectCallCoverage(Coverage):
      def _start(self):
          # 替换默认的trace函数
          self._trace = self.custom_trace
          super()._start()
  
      def custom_trace(self, frame, event, arg):
          if event == 'line':
              # 获取调用栈，取直接调用者（栈的第1层，当前执行帧是第0层）
              stack = inspect.stack()
              if len(stack) > 1:
                  caller_func = stack[1].function
                  # 判断调用者是否为测试函数
                  if caller_func.startswith('test_'):
                      # 记录该行到覆盖数据
                      self.data.add_line(frame.f_code.co_filename, frame.f_lineno)
          return self.custom_trace
  

  之后在pytest配置中使用这个自定义的Coverage类即可。

二、后处理Coverage报告 #

如果不想改动Coverage的核心逻辑，可以先正常生成覆盖数据，再通过脚本过滤掉嵌套调用的代码行：

1. 先运行pytest --cov得到原始覆盖数据（比如导出为coverage.json）。
2. 用Python自带的ast模块或者静态分析工具，解析代码找出每个测试函数直接调用的方法/函数。
3. 遍历原始覆盖数据，只保留这些直接调用方法内的代码行，将嵌套调用方法的代码行从覆盖结果中移除。
4. 用处理后的覆盖数据重新生成报告。

这种方式的关键是准确识别测试函数的直接调用关系，通过AST分析测试函数内的调用语句，就能对应到具体的目标方法。

三、结合Pytest钩子做动态过滤 #

利用Pytest的钩子函数（比如pytest_runtest_call），在每个测试函数执行前后记录当前测试函数的信息，然后在Coverage收集数据时做过滤：

• 在pytest_runtest_call钩子中，把当前执行的测试函数名存入全局变量。
• 在Coverage的trace函数中，检查当前代码行的调用栈，只有当直接调用者是全局变量中记录的测试函数时，才标记该行已覆盖。

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