使用装饰器进行数据驱动测试

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