使用Decorator进行数据驱动测试

Question

我试图用decorator参数化一个函数。最后，我能让它运行如预期，在许多打击和试验。但我仍然不满意，好像它是有效的，这似乎不是正确的方法。在

请帮助我改进这个代码。在

这是我的代码：

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) 

以下是更新后的代码：

^{pr2}$

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