如何使用pytest测试numba向量化函数并解决覆盖率统计缺失问题

核心原因 #

这不是你代码的问题，是覆盖率统计工具和numba编译机制的兼容性问题。
pytest-cov 底层依赖 coverage.py 实现覆盖率统计，原理是挂钩Python解释器的字节码执行流程，对应代码行的字节码被执行就会被统计为命中。而 @vectorize、@guvectorize 这类numba装饰器会在模块导入阶段，就把你写的Python函数逻辑编译为优化后的机器码，实际调用时直接走编译后的机器码路径，原Python函数的字节码完全不会被执行，因此coverage无法检测到这些代码行被调用过。

解决办法 #

你可以根据自己的需求选择以下两种方案：

• 测试时临时关闭numba JIT编译
  执行测试命令前加上环境变量 NUMBA_DISABLE_JIT=1，完整命令如下：

NUMBA_DISABLE_JIT=1 pytest --cov --cov-report term-missing

这种模式下numba装饰器会直接返回原始Python函数，走正常Python执行流程，coverage就能正常统计覆盖率。该模式只会关闭numba的性能优化，不会影响函数功能逻辑，测试结果的正确性不受影响。

• 配置coverage规则排除numba装饰的函数
  如果你已经确认函数逻辑正确，不需要统计这部分覆盖率，可以在项目根目录的 .coveragerc 配置文件中添加规则，将所有被numba装饰的函数排除在覆盖率统计范围之外，避免无意义的覆盖率缺口提示。

补充说明 #

你提供的测试用例逻辑完全正确，运行测试时所有逻辑分支都已经被覆盖，只是覆盖率工具无法检测到numba编译后的函数执行情况，不需要修改测试逻辑。
内容的提问来源于stack exchange，提问作者alegiac