pytest参数化详解

迈不过友情╰ 2024-04-01 12:35 96阅读 0赞

![在这里插入图片描述][b94bdd795bd9475b83d4874bb67f1b89.jpeg_pic_center]

--------------------

#### 文章目录 ####

*  一.概念
 *   *  pytest有三种传参方式：
 *  二.单个参数
 *  三.多个参数
 *   *  案例1
     *  案例2:使用py文件存放测试数据
     *  案例3:使用yaml文件存放测试数据
     *  案例4:使用json文件存放测试数据
 *  四.对测试类参数化
 *   *  案例1

## 一.概念 ##

参数化，就是把测试过程中的数据提取出来，通过参数传递不同的数据来驱动用例运行。其实也就是数据驱动的概念。

可以使用 @pytest.mark.parametrize(argnames, argvalues) 装饰器达到批量传送参数的目的

在 unittest 中，使用ddt库配合unittest实现数据驱动。在pytest中并不需要额外的库，通过pytest.mark.parametrize()即可实现参数化。

parametrize()的第一个参数是用逗号分割的字符串列表，第二个参数是一个值列表

### pytest有三种传参方式： ###

> @pytest.mark.parametrize() 通过装饰器方式进行参数化（最常使用）  
> pytest.fixture()方式进行参数化，fixture装饰的函数可以作为参数传入其他函数  
> conftest.py文件中存放参数化函数，可作用于模块内的所有测试用例

## 二.单个参数 ##

test\_params

import pytest
    
    def add(a,b):
        return a+b
    
    class TestParams:
    
        @pytest.mark.parametrize('a',[1,2,3,4,5])
        def test_par1(self,a):
            assert add(a,1)==a+1

命令行执行:

pytest

![在这里插入图片描述][1010ccfb69a443b49351fb63fb329d1f.png]  
特别注意:  
@pytest.mark.parametrize() 装饰器接收两个参数，  
第一个参数是以字符串的形式标识用例函数的参数，  
第二个参数以列表或元组的形式传递测试数据。  
![在这里插入图片描述][7051d59666084970be159d67d8a89246.png]

## 三.多个参数 ##

### 案例1 ###

test\_params

import pytest
    
    def add(a,b):
        return a+b
    
    class TestParams:
    
        @pytest.mark.parametrize('a,b,c',[[1,2,3],[3,4,5],[1,3,4],[2,5,7]])
        def test_par1(self,a,b,c):
            assert add(a,b)==c

执行命令  
![在这里插入图片描述][7fd262e8a6094d72b35a854bffbb25ac.png]  
特别注意:  
多个参数之间要用逗号分隔  
参数名称和个数要一一对应

![在这里插入图片描述][d135c5374e314e77beeafdf11a9af87f.png]

### 案例2:使用py文件存放测试数据 ###

测试数据

login_data_list = [
        	{
        'username': 'kobe', 'password': 666666, 'expect': '登录成功'},
        	{
        'username': 'kobe', 'password': 111111, 'expect': '用户名或者密码错误'},
        	{
        'username': '', 'password': 666666, 'expect': ''},
    	]

test\_work

import pytest
    from data.data1 import login_data_list
    
    class TestParams:
    
        @pytest.mark.parametrize('test_data',login_data_list)
        def test_par1(self,test_data):
            username=test_data['username']
            password=test_data['password']
            assert username=='kobe' and password==666666

测试结果  
![在这里插入图片描述][5e2201b866b64ce685ced1211ab40f78.png]

### 案例3:使用yaml文件存放测试数据 ###

![在这里插入图片描述][6160212836c34f519fd455d634d3086f.png]

yaml文件

-
      name: '账号密码正确'
      username: 'kobe'
      password: 666666
    
    -
      name: '用户名为空'
      username:
      password: 666666
    
    -
      name: '用户名或者密码错误'
      username: 'james'
      password: 666666

read\_yaml.py

import os
    import yaml
    
    def read_yaml(yaml_file_path):
        try:
            with open( yaml_file_path, "r",encoding="utf-8") as f:
                value = yaml.load(stream=f, Loader=yaml.FullLoader)
        except:
            with open(yaml_file_path, "r",encoding="gbk") as f:
                value = yaml.load(stream=f, Loader=yaml.FullLoader)
        print(value)
        return value
    
    
    
    if __name__ == '__main__':
        read_yaml(r'D:\project_development\api_pytest\data\test_yaml.yaml')

读取结果：

[{
        'name': '账号密码正确', 'username': 'kobe', 'password': 666666}, {
        'name': '用户名为空', 'username': None, 'password': 666666}, {
        'name': '用户名或者密码错误', 'username': 'james', 'password': 666666}]

test\_yaml\_work.py

import pytest
    from utils.read_yaml import read_yaml
    
    class TestParams:
    
        @pytest.mark.parametrize('case',read_yaml(r'D:\project_development\api_pytest\data\test_yaml.yaml'))
        def test_login(self,case):
            username=case['username']
            password=case['password']
            assert username=='kobe' and password==666666

执行测试：

D:\project_development\api_pytest>pytest testcases/params/test_yaml_work.py

![在这里插入图片描述][ef9c0cb262b24c6ebcf4ab8bd5fc79b9.png]

### 案例4:使用json文件存放测试数据 ###

![在这里插入图片描述][fa97a0ff8e1e43bcad4cabb78c58f152.png]

test\_json.json

[
      {
        
        "name": "账号密码正确",
        "username": "kobe",
        "password": 666666
      },
      {
        
        "name": "用户名为空",
        "username": "",
        "password": 666666
      },
      {
        
        "name": "用户名或者密码错误",
        "username": "james",
        "password": 666666
      }
    ]

read\_json.py

import os
    import json
    
    def read_json(json_file_path):
        try:
            with open(json_file_path, "r", encoding="utf-8") as f:
                datas = json.load(f)
        except:
            with open(json_file_path, "r", encoding="gbk") as f:
                datas = json.load(f)
        if not isinstance(datas, list):
            raise ValueError('json文件内的用例数据格式不符护规范')
    
        print(datas)
        return datas
    
    
    if __name__ == '__main__':
        read_json(r'D:\project_development\api_pytest\data\test_json.json')

读取结果：

[{
        'name': '账号密码正确', 'username': 'kobe', 'password': 666666}, {
        'name': '用户名为空', 'username': '', 'password': 666666}, {
        'name': '用户名或者密码错误', 'username': 'james', 'password': 666666}]

test\_json\_work.py

## 四.对测试类参数化 ##

测试类的参数化，其实际上也是对类中的测试方法进行参数化。类中的测试方法的参数必须与@pytest.mark.parametrize()中的标识的参数个数一致。

![在这里插入图片描述][236f2ec5f56547458ade2519dce347de.png]

### 案例1 ###

import pytest
    
    def add(a,b):
        return a+b
    
    
    @pytest.mark.parametrize('a,b,c',[[1,2,3],[1,3,4],[2,5,7]])
    class TestParams:
    
        def test_par1(self,a,b,c):
            assert add(a,b)==c
    
        def test_par2(self,a,b,c):
            assert add(a,b)==c

执行结果

![在这里插入图片描述][381d27c92ac04025a9753d3297cc2839.png]

--------------------

![在这里插入图片描述][8af8b3a37def40c692605cd92d2f6135.jpeg_pic_center]

[b94bdd795bd9475b83d4874bb67f1b89.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/12ec22b0d1e445d187e4990c80347659.jpeg
[1010ccfb69a443b49351fb63fb329d1f.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/8ac6056b749a4ca5b59ecaa1ff962304.png
[7051d59666084970be159d67d8a89246.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/732f092c3f3a4681aba809f93a13c95d.png
[7fd262e8a6094d72b35a854bffbb25ac.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/9cf4130b30284a0286b964553dc6c5ab.png
[d135c5374e314e77beeafdf11a9af87f.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/5410ef6cd2d1461bb3de4615dab20490.png
[5e2201b866b64ce685ced1211ab40f78.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/ed9dcd6afec340b2983ccd3e59580604.png
[6160212836c34f519fd455d634d3086f.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/f16a40c4ac8d4200aa916176876ee21d.png
[ef9c0cb262b24c6ebcf4ab8bd5fc79b9.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/8c295c38ddd848dfa1d1194199f7ccb3.png
[fa97a0ff8e1e43bcad4cabb78c58f152.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/dfc36bc057564eafa337565f0a1fc409.png
[236f2ec5f56547458ade2519dce347de.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/13d71dbb01e64f4fa5130acbc56ea03e.png
[381d27c92ac04025a9753d3297cc2839.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/ce92f3324f974c20b102f38d2cd34ce7.png
[8af8b3a37def40c692605cd92d2f6135.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/7a98aa1dd41f42a781dbec72a9621535.jpeg