Python:如何在单元(nose)测试期间忽略修饰符?

仅对于NUBA,您可以设置环境变量(例如。 export NUMBA_DISABLE_JIT=1 )使jit装饰器无需操作。

http://numba.pydata.org/numba-doc/dev/user/troubleshoot.html#disabling-jit-compilation

您可能可以欺骗nose使用自己的装饰器来保持函数不变。 写入文件 numba.py 包含此函数的:

def jit(func):
    return func

放在你的蟒蛇身上。 可以从该文件所在的目录运行测试,也可以在中执行此操作 您的测试文件作为第一行:

import sys

sys.path.insert(0, 'path/to/dir/with/myfile') 

或者,可以设置环境变量 PYTHONPATH 到文件所在的目录。

窗户:

set PYTHONPATH=%PYTHONPATH%;path/to/dir/with/myfile

Unix/Mac:

export PYTHONPATH=$PYTHONPATH$:path/to/dir/with/myfile

您可以使用新的装饰器修补装饰器,您可以控制:

import functools

def patch(parent, obj_name, switch_reg, switch_name):
    def gen_switcher():
        def wrapper(func):
            ori_wrapped = ori_decorator(func)
            @functools.wraps(func)
            def _(*args, **kwargs):
                if switch_reg.get(switch_name, False):
                    func_to_call = func
                else:
                    func_to_call = ori_wrapped
                print(func_to_call)
                return func_to_call(*args, **kwargs)
            return _
        return wrapper
    ori_decorator = getattr(parent, obj_name)
    setattr(parent, obj_name, gen_switcher())

使用:

# have to patch the decorator before applying it
import numba
switchs = {}
patch(numba, 'jit', switchs, 'DISABLE_NUMBA_JIT')

@numba.jit
def f(a, b):
  return a + b

f(1, 2)

产量:

CPUDispatcher(<function f at 0x10a5f90d0>)

然后使用:

# this part could be rewrited as a context manager    
switchs['DISABLE_NUMBA_JIT'] = True
f(1, 2)

获取:

<function f at 0x10a5f90d0>

如果您正在使用 coverage ,您可以修改CI .yml 通过添加文件 NUMBA_DISABLE_JIT=1 就在你的 新闻报道 命令:

NUMBA_DISABLE_JIT=1 coverage run -p "test_*.py"

我会将该函数分离为一个非修饰部分,由您进行单元测试,然后让您的实际函数具有修饰器,只需调用helper函数:

@numba.jit
def f(a, b):
  return f_undecorated(a, b)

def f_undecorated(a, b):
  return a + b

仅为编写单元测试 f_undecorated .