驭风万里无垠

Python中根据不同参数组合产生单独的test case的一种方法

 
Python自带的unittest和test两个模块为编写test case提供了很灵活的支持,最常用的情况就是继承自unittest.TestCase类,然后对每一个要进行测试的行为写一个test_开头的类成员函数,最后可以利用test.test_support.run_unittest函数跑所有的test case.
 
 
在某种情况下,可能需要用不同的参数组合测试同样的行为,这些行为要么很耗时间(譬如下载数据),要么是你希望从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成员函数的调用(只需要在实际执行前就可以,可以放在模块中自动执行亦可以手动调用)

最后的代码就有了:

 
01 import unittest
02 from test import test_support
03
04 def MyTestCase(unittest.TestCase):
05     def setUp(self):
06         #some setup code
07         pass
08        
09     def clear(self):
10         #some cleanup code
11         pass
12        
13     def action(self, arg1, arg2):
14         pass
15        
16     @staticmethod   
17     def getTestFunc(arg1, arg2):
18         def func(self):
19             self.actions(arg1, arg2)
20         return func
21        
22 def __generateTestCases():
23     arglists = [('arg11', 'arg12'), ('arg21', 'arg22'), ('arg31', 'arg32')]
24     for args in arglists:
25         setattr(MyTestCase, 'test_func_%s_%s'%(args[0], args[1]),
26             MyTestCase.getTestFunc(*args) )
27 __generateTestCases()
28      
29 def test_main():
30     test_support.run_unittest(MyTestCase)

 

如此,添加一个新的可变参数组合,就会新生成一个test case, 只需要将参数组合添加到arglist中就可以了。

posted on 2009-10-12 21:20 skyscribe 阅读(1573) 评论(0)  编辑 收藏 引用


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


<2009年12月>
293012345
6789101112
13141516171819
20212223242526
272829303112
3456789

导航

统计

常用链接

留言簿(3)

随笔分类

随笔档案

搜索

最新评论

阅读排行榜

评论排行榜