python利用ddt做数据驱动

最新推荐文章于 2025-05-17 19:53:12 发布

原创最新推荐文章于 2025-05-17 19:53:12 发布 · 605 阅读

1 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

77 篇文章

订阅专栏

本文介绍了一个Python函数用于从CSV文件中读取数据，并展示了如何在unittest框架下使用DDT装饰器进行数据驱动的单元测试。通过读取CSV文件的数据并将其作为测试用例输入，可以有效地进行应用中心模块的接口测试。

读取csv文件的函数

def getCsvData(file):
    # 读取CSV文件
    value_rows = []
    base_dir = str(os.path.dirname(os.path.dirname(__file__)))
    base_dir = base_dir.replace('\\', '/')
    csv_path = base_dir + file   # csv路径
    with open(csv_path) as f:   # , encoding='UTF-8'
        f_csv = csv.reader(f)
        next(f_csv)
        for r in f_csv:
            value_rows.append(r)
    print(value_rows)
    return value_rows

@ddt
class RyomsTest(unittest.TestCase):
    """大运营模块测试"""
    def setUp(self):
        pass

    def tearDown(self):
        pass

    @data(*getCsvData("/data/ryoms_app_center.csv"))
    @unpack   # 解包
    # @file_data("/data/ryoms_app_center.csv")
    def test_easy_smart_basic_module_mall(self, data, alert_data):
        """应用中心-参数配置-商场配置-获取商场配置接口/easy-smart-basic/module/mall"""
        print(data)
        url = env_url + "/easy-smart-basic/module/mall"
        data = eval(data)  # 去除csv带来的""引号
        r = requests.post(url, headers=header, json=data)
        print(r.status_code)
        print(r.json())
        result = r.json()
        self.assertEqual(r.status_code, 200)
        self.assertEqual(result['meta']['msg'], alert_data)
        self.assertEqual(result['success'], True)