使用裝飾器進行數據驅動測試

Question

我想參數化使用裝飾器的函數。最後，我能夠在經過大量命中和試驗之後按預期運行。但是我仍然不滿意，就好像它在工作，但似乎並不是正確的做法。

請幫我改進這段代碼。

這裏是我的代碼：

def WarmWelcome(fn): 
    def wrapped(DataProvider): 
     for name in DataProvider(): 
     print fn(name) + ":)" 
    return wrapped 

def DataProvider(): 
    names=["abc","xyz","def"] 
    for name in names: 
     yield name 

@WarmWelcome 
def hello(name): 
    return "hello " +name 

hello(DataProvider) 

下面是更新後的代碼：

def WarmWelcome(DataProvider): 
    def real_decorator(fn): 
    def wrapped(): 
     for name in DataProvider(): 
     print fn(name) + ":)" 
    return wrapped 
    return real_decorator 

def DataProvider(): 
    names=["abc","xyz","def"] 
    for name in names: 
     yield name 
@WarmWelcome(DataProvider) 
def hello(name): 
    return "hello " +name 

hello() 

Answer 1

也有可能提供的數據集權WarmWelcome裝飾：

def WarmWelcome(*data_sets): 
    def _decorator(func): 
     def _func(): 
      for ds in data_sets: 
       func(*ds) 
     return _func 
    return _decorator 

@WarmWelcome(
    ("abc",), 
    ("xyz",), 
    ("def",), 
) 
def hello(name): 
    return "hello " +name 

原文：PHPUnit-like dataProvider implementation for Python unittest