python中自带的三个装饰器_python学习：三个测试库的装饰器实现思路

在 Python 中实现参数化测试的几个库，并留下一个问题：

它们是如何做到把一个方法变成多个方法，并且将每个方法与相应的参数绑定起来的呢？

我们再提炼一下，原问题等于是：在一个类中，如何使用装饰器把一个类方法变成多个类方法(或者产生类似的效果)？

# 带有一个方法的测试类

class TestClass:

def test_func(self):

pass

# 使用装饰器，生成多个类方法

class TestClass:

def test_func1(self):

pass

def test_func2(self):

pass

def test_func3(self):

pass

Python 中装饰器的本质就是移花接木，用一个新的方法来替代被装饰的方法。在实现参数化的过程中，我们介绍过的几个库到底用了什么手段/秘密武器呢？

1、ddt 如何实现参数化？

先回顾一下上篇文章中 ddt 库的写法：

import unittest

from ddt import ddt,data,unpack

@ddt

class MyTest(unittest.TestCase):

@data((3, 1), (-1, 0), (1.2, 1.0))

@unpack

def test(self, first, second):

pass

ddt 可提供 4 个装饰器：1 个加在类上的 @ddt，还有 3 个加在类方法上的 @data、@unpack 和 @file_data(前文未提及)。

先看看加在类方法上的三个装饰器的作用：

# ddt 版本(win)：1.2.1

def data(*values):

global index_len

index_len = len(str(len(values)))

return idata(values)

def idata(iterable):

def wrapper(func):

setattr(func, DATA_ATTR, iterable)

return func

return wrapper

def unpack(func):

setattr(func, UNPACK_ATTR, True)

return func

def file_data(value):

def wrapper(func):

setattr(func, FILE_ATTR, value)

return func

return wrapper

它们的共同作用是在类方法上 setattr() 添加属性。至于这些属性在什么时候使用？下面看看加在类上的 @ddt 装饰器源码：