Python中unittest 用不同的参数组合，测试同样的行为；使test case根据参数多次运行

最新推荐文章于 2025-01-08 21:02:44 发布

转载最新推荐文章于 2025-01-08 21:02:44 发布 · 5.5k 阅读

selenium 专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种使用Python unittest框架进行参数化测试的方法。通过自动生成针对不同参数组合的测试案例，提高了测试效率与可读性。文章提供了一个具体示例，展示了如何实现这一功能。

原文：http://www.2cto.com/kf/201501/367270.html

在某种情况下，需要用不同的参数组合，测试同样的行为，我们希望从test case的执行结果上知道在测试什么，而不是单单得到一个大的 test case；

此时如果仅仅写一个test case并用内嵌循环来进行，那么其中一个出了错误，很难从测试结果里边看出来问题

所以现在问题的关键在于是否有办法根据输入参数的不同组合产生出对应的test case？

譬如你有10组数据，那么得到10个test case，这种情况当然不适用纯手工的方式写那么多个test_成员函数。

一种可能的思路是不利用unittest.TestCase这个类框中的test_成员函数的方法，而是自己写runTest这个成员函数，那样会有一些额外的工作，而且看起来不是那么“智能”。

那该如何让框架自动调用testcase呢？

我们的思路是：利用setattr来自动为已有的TestCase类添加成员函数为了使这个方法凑效，需要用类的static method来生成decorate类的成员函数，并使该函数返回一个test函数对象出去在某个地方注册这个添加test成员函数的调用(只需要在实际执行前就可以，可以放在模块中自动执行亦可以手动调用)

dome：

# !/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest


class TestCases(unittest.TestCase):

    def setUp(self):
        pass

    def action(self, arg1, arg2):
        print arg1, arg2

    @staticmethod
    def getTestFunc(arg1, arg2):
        def func(self):
            self.action(arg1, arg2)
        return func

def __generateTestCases():
    arglists = [('arg11', 'arg12'), ('arg21', 'arg22'), ('arg31', 'arg32')]
    for args in arglists:
        setattr(TestCases, 'test_func_%s_%s' % (args[0], args[1]),
                TestCases.getTestFunc(*args))
__generateTestCases()

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