python ddt 数据驱动测试

测试积点老人

1. # -*- coding: utf-8 -*-
   
2. import unittest
   
3. from ddt import ddt, data, unpack
   
4. import csv
   
5. from collections import namedtuple
   
6. 
   
7. 
   
8. def add(a,b):
   
9.     c = a+b
   
10.     return c
    
11. 
    
12. 
    
13. def addstr(a,b):
    
14.     c = a + b
    
15.     return c
    
16. 
    
17. 
    
18. def get_csv_data():
    
19.     value_rows = []
    
20.     with open('./mfile.csv') as f:      
    
21.         f_csv = csv.reader(f)
    
22.         next(f_csv)
    
23.         for r in f_csv:
    
24.             value_rows.append(r)
    
25.             
    
26.     return value_rows
    
27.             
    
28. 
    
29. 
    
30. @ddt
    
31. class Test(unittest.TestCase):
    
32. 
    
33. 
    
34.     @data((1,1,2), (1,2,3))
    
35.     @unpack
    
36.     def test_addnum(self,a, b, expected_value):
    
37.         self.assertEqual(add(a,b),expected_value)
    
38.         
    
39.     @data(*get_csv_data())
    
40.     @unpack
    
41.     def test_addstr(self,a,b,expected_value):
    
42.         self.assertEqual(addstr(a,b), expected_value)
    
43.         
    
44. if __name__ == "__main__":
    
45.     suite = unittest.TestLoader().loadTestsFromTestCase(Test)
    
46.     unittest.TextTestRunner(verbosity=2).run(suite)
    
47. 
    
48. 
    
49. 
    
50. csv文件：
    
51. 
    
52. value1,value2,result
    
53. aa,bb,aabb
    
54. ww,r,wwr
    
55. 111,222,111222
    
56. 1,2,3
    

复制代码

运行结果：

1. test_addnum_1__1__1__2_ (__main__.Test) ... ok
   
2. test_addnum_2__1__2__3_ (__main__.Test) ... ok
   
3. test_addstr_1___aa____bb____aabb__ (__main__.Test) ... ok
   
4. test_addstr_2___ww____r____wwr__ (__main__.Test) ... ok
   
5. test_addstr_3___111____222____111222__ (__main__.Test) ... ok
   
6. test_addstr_4___1____2____3__ (__main__.Test) ... FAIL
   

复制代码