我可以在单元测试的设置阶段编译 numba 吗？答案

【问题标题】：Can I compile numba in the setup phase of a unit test?我可以在单元测试的设置阶段编译 numba 吗？
【发布时间】：2014-02-17 15:45:28
【问题描述】：

我使用 numba 使用 autojit 装饰器来加速我的部分 Python 代码。测试通过，%timeit 显示代码确实加速了。

第一次执行很慢，因为我认为 numba 正在编译代码。说得通。但是，当我运行一套测试时，它们非常运行得很慢，从我更改前大约 10 秒内的 10 次测试到 117 秒内的相同测试。似乎 numba 必须一次又一次地为每个测试单独编译。

我可以避免这种情况吗？我尝试在 setUp 函数中运行一个简单的测试，认为可能会为所有测试编译一次该函数，但这并没有显着改变运行时间。

【问题讨论】：

【解决方案1】：

这样的事情可能会奏效：

from numba import autojit

def autojit_except_when_unit_testing(func):
    if UNIT_TESTING:
        return func
    return autojit(func)

可能还会遇到 numba 的缓存问题，因为这是一个非常重要的用例。我通常会很犹豫在这样不同的环境中运行单元测试和生产代码，但是需要永远运行的单元测试不会经常运行。您几乎可以肯定也应该使用 numba 进行测试，只是频率要低一些。

【讨论】：