调用Cython模块后，如何获取后续代码行的覆盖率？

问题描述 #

调用第三方Cython库的函数后，后续自有Python代码明明已执行（可通过抛出异常验证测试会触发失败），但覆盖率工具始终标记这些代码为未执行。示例代码如下：

def some_function():
    cython_library.do_something()
    do_something_else()  # 覆盖率工具始终显示未执行

由于无法重新编译第三方Cython库以生成覆盖率信息，且不能完全mock该库（会失去测试意义），需要取巧的方法让覆盖率工具正确识别后续代码的执行。

可行解决方案 #

1. 插入轻量Python函数调用 #

定义一个空的辅助函数，在Cython调用后执行，强制Python解释器回到正常字节码追踪流程：

def _coverage_trigger():
    pass

def some_function():
    cython_library.do_something()
    _coverage_trigger()
    do_something_else()

2. 使用exec执行空语句 #

exec会触发Python解释器处理字节码，可恢复覆盖率工具的追踪：

def some_function():
    cython_library.do_something()
    exec("pass")
    do_something_else()

3. 手动调用覆盖率工具API记录（针对coverage.py） #

如果使用coverage.py作为覆盖率工具，可手动标记当前代码帧为已执行：

import coverage
import inspect

def some_function():
    cython_library.do_something()
    # 手动触发覆盖率记录
    cov = coverage.Coverage.current()
    if cov:
        frame = inspect.currentframe()
        cov.data.record(frame.f_code, frame.f_lineno)
    do_something_else()

原理说明 #

第三方Cython函数直接编译为机器码执行，会绕过Python解释器的字节码追踪机制，导致覆盖率工具无法感知后续代码的执行。上述方法通过插入纯Python操作，让解释器重新进入正常的字节码执行流程，从而让覆盖率工具正确记录后续代码的执行状态。

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