python单元测试的unittest读取csv文件

1.一个用来存放数据
2.一个用来读取那个存放数据的csv
3.一个用来执行测试脚本
4.一个用来生成html文件

注意点:
脚本里的内容需要都要测试通过才能成功…
比如要执行三个脚本,有一个脚本不成功,就会测试失败.

1.存放数据

// 创建一个csv文件输入数据
1,2,3
4,5,9

2.读取存放的csv数据

// 创建一个py文件,编写读取csv的脚本
# 首先导入csv模块
import csv
# 创建类
class ReadCsv():
    def read_csv(self):
        item = []  # 定义一个空列表
        c = csv.reader(open("../datademo/testcsv.csv", "r"))  # 得到csv文件对象
        # for循环这个列表
        for csv_i in c:
            item.append(csv_i)  # 将获取的数据添加到列表中
        return item

#     实例化类,打印值
readcsv = ReadCsv()
print(readcsv.read_csv())

3.执行测试脚本

// 创建一个py文件,执行脚本
#   导入unittest包
import unittest
#   引用读取csv文件的类
from readdemo.readcsv import ReadCsv

# 实例化读取文件的类
readscv = ReadCsv()
from demo.kaifa import KaiFa

# 获取类中的列表
lists = readscv.read_csv()
kaifa = KaiFa()

# 创建类
class TestCsv(unittest.TestCase):
    # 创建方法
    def test_csv(self):
        # for循环这个列表
        for csv_i in lists:
            # 定义一个值=开发的方法
            zhi = kaifa.add(int(csv_i[0]), int(csv_i[1]))
            # 比较值
            self.assertEqual(zhi, int(csv_i[2]))

# 运行
if __name__ == '__main__':
    unittest.main()

4.生成html文件

// 运行生成html的脚本
import unittest

from datademo.HTMLTestRunner import HTMLTestRunner
from test.testcsv import TestCsv

class HtmlClass():
    def shengchenghtml(self):
        suite = unittest.TestSuite()
        caselist = ["test_csv"]
        for case in caselist:
            suite.addTest(TestCsv(case))

        with open("../csvtest.html", "wb") as f:
            HTMLTestRunner(
                stream=f,
                title="单元测试",
                description="测试csv",
                verbosity=2
            ).run(suite)

if __name__ == '__main__':
    test = HtmlClass()
    test.shengchenghtml()

结果:

html: