Python unittest & ddt 实现
ddt 对于测试来说,非常方便生成用例,特别是数据很多,用例类似的。
个人经历的一个项目,文本过滤。需要对海量数据进行排查过滤文字,就是使用这个方式处理。
一、安装 unittest & ddt
pip install unittest
pip install ddt
二、ddt 实现
- 核心代码
# coding: utf-8
import unittest
import ddt
from xmlrunner import XMLTestRunner
from Predict import Predict
from char_cnn_api import get_csv_data, write_csv_data
@ddt.ddt
class CharCnnTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.test_res = {
'matched': [], 'unmatched': [], 'results': []}
#数据源获取
@ddt.data(*get_csv_data(