Python：ユニット（鼻）テスト中にデコレータを無視する方法は？

Question

私のコード上の特定のデコレータを無視して、関連する機能をスキップしたいのですが、単体テストの一部として実行している場合に限ります。これはできますか？例えば

私はそうのようなnumbaデコレータで定義され、関数fを持っている：私はnumbaのマジックが起こることを望んでいない上記の関数を呼び出す単体テストを実行すると

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

、などいくつかのプラットフォームでは処理が遅くなり、エラーが発生する可能性があります。 numbaのジャストインタイム（および/またはすべてのnumba）デコレータを適用せずにテストを実行するように鼻に指示する場所を設定することは可能ですか？

Answer 1

のみ、あなたが（例えば、環境変数を設定することができます。 export NUMBA_DISABLE_JIT=1）を使ってジットデコレータのno-opsを作成します。

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

Answer 2

機能を変更しない独自のデコレータを使用すると、鼻を騙すことができます。

def jit(func): 
    return func 

とあなたのPYTHONPATHに置か： この機能が含まれているファイルnumba.pyを書きます。 いずれかが、このファイルが置かれているディレクトリからテストを実行したり、最初の行として でテストファイルを次の操作を行います。

import sys 

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

また、あなたがあなたのファイルがあるディレクトリに環境変数PYTHONPATHを設定することができます。

のWindows：

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

のUnix/Macの場合：

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

Answer 3

あなたはあなたがコントロールすることができ、新たなデコレータとデコレータにパッチを適用できますと

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) 

GET：numbaについては

<function f at 0x10a5f90d0> 

Answer 4

私はあなたのユニットテスト一部非飾らに機能を分離う - と、あなたの本当の機能をするデコレータを持っており、単にヘルパー関数を呼び出す：

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

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

書き込みユニットテストはf_undecoratedです。