Python DDT模块详解：@data与@unpack装饰器的数据驱动用法

巷口听风

发布时间：2026-10-13| 4人浏览过

原创

本文深入解析Python中ddt模块的核心功能，重点讲解@data和@unpack装饰器的区别与用法。通过具体代码示例，说明如何直接将测试数据传入用例，以及如何利用@unpack分解元组或列表数据以匹配多个参数。文章还分析了未使用@unpack时可能引发的TypeError错误，帮助开发者正确实现数据驱动自动化测试。

ddt 是一个第三方 Python 模块，使用前需要通过命令行执行 pip install ddt 进行安装。

DDT 库主要包含一个类装饰器 ddt 以及两个方法装饰器：data 和 unpack。其中，data 装饰器用于直接指定传递给测试用例的数据源。

在常规使用场景中，@data 装饰器内的数据通常作为单一参数传递给测试函数。如果 @data 中包含多个数据项（如元组、列表或字典），开发者需要在脚本中手动分解这些数据，或者使用 @unpack 装饰器来自动分解数据。

例如使用语法 @data(a,b)。

此时，装饰器会生成两个独立的测试用例，分别使用 a 和 b 作为参数各运行一次测试。

例如使用语法 @data([a,d],[c,d])。

若未添加 @unpack 装饰器，整个列表 [a,b] 将被视为单个参数传入测试函数执行。

若添加了 @unpack 装饰器，列表 [a,b] 会被分解为独立元素，并分别按顺序传递给测试函数中的对应参数。

具体实现细节请参考以下代码示例。

import unittest
from ddt import ddt,data,unpack

@ddt
class MyTesting(unittest.TestCase):
  def setUp(self):
    print('this is the setUp')
  @data([1,2,3])
  def test_1(self,value):
    print(value)

  @data([3,2,1],[5,3,2],[10,4,6])
  @unpack
  def test_minus(self,a,b,expected):
    actual = int(a) - int(b)
    expected = int(expected)
    self.assertEqual(actual, expected)

  @data([2,3],[4,5])
  def test_compare(self,a,b):
    self.assertEqual(a,b)

  def tearDown(self):
    print('this is tearDown')

if __name__ == '__main__':
  unittest.main(verbosity=2)

对运行结果的分析如下。

1. test_1 的测试结果为 ok。这是因为 [1,2,3] 作为一个整体列表被传递给了参数 value，因此 value 打印出的值为 [1,2,3]。

test_1_1__1__2__3_ (__main__.MyTesting) ... ok
test_compare_1__2__3_ (__main__.MyTesting) ... ERROR
[1, 2, 3]

2. test_minus 的测试结果同样为 ok。由于在 @data(...) 下方添加了 @unpack 装饰器，数据会被分解处理，最终生成 3 组独立的测试数据，分别为：

1.[3,2,1]
2.[5,3,2]
3.[10,4,6]
test_minus_1__3__2__1_ (__main__.MyTesting) ... ok
test_minus_2__5__3__2_ (__main__.MyTesting) ... ok
test_minus_3__10__4__6_ (__main__.MyTesting) ... ok

3. test_compare 的测试结果为 fail。原因是未添加 @unpack 装饰器，尽管系统仍将其识别为 2 组测试数据，但 [2,3] 作为一个整体列表被传递给了参数 a，导致参数 b 没有接收到任何值。因此，执行时抛出 TypeError: test_compare() missing 1 required positional argument: 'b' 错误。

test_compare_1__2__3_ (__main__.MyTesting) ... ERROR
test_compare_2__4__5_ (__main__.MyTesting) ... ERROR

this is the setUp
ERROR: test_compare_1__2__3_ (__main__.MyTesting)
this is tearDown
----------------------------------------------------------------------
Traceback (most recent call last):
 File "D:\python\lib\site-packages\ddt.py", line 139, in wrapper
  return func(self, *args, **kwargs)
TypeError: test_compare() missing 1 required positional argument: 'b'

======================================================================
ERROR: test_compare_2__4__5_ (__main__.MyTesting)
----------------------------------------------------------------------
Traceback (most recent call last):
 File "D:\python\lib\site-packages\ddt.py", line 139, in wrapper
  return func(self, *args, **kwargs)
TypeError: test_compare() missing 1 required positional argument: 'b'

以上是对 ddt 模块的学习总结。ddt 还提供了 file_data 方法，支持从 json 或 yaml 文件中读取测试数据，相关内容将在后续文章中继续介绍。