51Testing软件测试论坛

标题: python ddt 数据驱动测试 [打印本页]
作者: 测试积点老人    时间: 2018-12-12 14:15
标题: python ddt 数据驱动测试
  1. # -*- coding: utf-8 -*-
  2. import unittest
  3. from ddt import ddt, data, unpack
  4. import csv
  5. from collections import namedtuple


  8. def add(a,b):
  9.     c = a+b
  10.     return c


  13. def addstr(a,b):
  14.     c = a + b
  15.     return c


  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.             


  30. @ddt
  31. class Test(unittest.TestCase):


  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)



  50. csv文件：

  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
复制代码






欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2