Могу ли я скомпилировать numba на этапе настройки модульного теста?

Question

Я использовал numba для ускорения частей моего кода Python с помощью декоратора autojit. Тест проходит и% timeit показывает, что код действительно ускоряется.

Первое исполнение выполняется медленно, поскольку, я полагаю, numba компилирует код. Имеет смысл. Но когда я запускаю набор тестов, они быстро запускают , из 10 тестов за 10 секунд до моих изменений в те же тесты за 117 секунд. Кажется, что numba необходимо компилировать снова и снова, отдельно для каждого теста.

Могу ли я избежать этого? Я попытался запустить один простой тест в функции setUp, думая, что может скомпилировать функцию там один раз для всех тестов, но это существенно не изменило время выполнения.

Answer 1

Нечто подобное может работать:

from numba import autojit 

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

Возможно также врезаться вопросы Numba в отношении кэширования, так как это очень важный случай использования. Я обычно был бы довольно нерешительным, чтобы запускать модульные тесты и производственный код в таких разных средах, но модульные тесты, которые выполняются навсегда, не запускаются так часто. Вы должны почти наверняка тестировать с numba также, реже.