python ddt 数据驱动测试

# -*- coding: utf-8 -*-
import unittest
from ddt import ddt, data, unpack
import csv
from collections import namedtuple


def add(a,b):
    c = a+b
    return c


def addstr(a,b):
    c = a + b
    return c


def get_csv_data():
    value_rows = []
    with open('./mfile.csv') as f:       
        f_csv = csv.reader(f)
        next(f_csv)
        for r in f_csv:
            value_rows.append(r)
            
    return value_rows
            


@ddt
class Test(unittest.TestCase):


    @data((1,1,2), (1,2,3))
    @unpack
    def test_addnum(self,a, b, expected_value):
        self.assertEqual(add(a,b),expected_value)
        
    @data(*get_csv_data())
    @unpack
    def test_addstr(self,a,b,expected_value):
        self.assertEqual(addstr(a,b), expected_value)
        
if __name__ == "__main__":
    suite = unittest.TestLoader().loadTestsFromTestCase(Test)

    unittest.TextTestRunner(verbosity=2).run(suite)

csv文件：

value1,value2,result
aa,bb,aabb
ww,r,wwr
111,222,111222
1,2,3

运行结果：

test_addnum_1__1__1__2_ (__main__.Test) ... ok
test_addnum_2__1__2__3_ (__main__.Test) ... ok
test_addstr_1___aa____bb____aabb__ (__main__.Test) ... ok
test_addstr_2___ww____r____wwr__ (__main__.Test) ... ok
test_addstr_3___111____222____111222__ (__main__.Test) ... ok
test_addstr_4___1____2____3__ (__main__.Test) ... FAIL