【三：pytest中装饰器parametrize实现数据驱动的使用】

一：pytest装饰器parametrize

@pytest.mark.parametrize(参数名，数据（list，tuple，字典列表，字典元祖）)

第一种基本用法：

import pytest
class TestApi:
#('name,age',[['百里',10],['一新',11],['娜扎',12]])
    @pytest.mark.parametrize('name',['张杰','谢娜','胡歌'])
    def test_api2(self,name):
        print(name)


if __name__ == '__main__':
    pytest.main(['-vs','02-数据驱动.py'])

第二种用法：

class TestApi:
#('name,age',[['百里',10],['一新',11],['娜扎',12]])
    @pytest.mark.parametrize('name,age',[['张杰',10],['胡歌',11],['娜扎',12]])
    def test_api2(self,name,age):
        print(name,age)


if __name__ == '__main__':
    pytest.main(['-vs','02-数据驱动.py'])

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TmLtr5IO-1721979295437)(https://i-blog.csdnimg.cn/blog_migrate/181d53ed3c596e9cae535fcfbb883657.png)]

二：pytest中的parametrize使用扩展

使用yaml+parametrize实现数据驱动

思路步骤：

• 第一步：准备yaml用例的测试文件
• 第二步：编写读取yaml文件的方法
• 第三步：编写测试用例的方法test_api2
• 第四步：在方法的前面加上实现数据驱动的装饰器parametrize，并以读取的方式获取数据

import pytest
import yaml


def read_testcase_yaml(yaml_path):
    with open(yaml_path, mode='r', encoding='utf-8')as f:
        value = yaml.load(stream=f, Loader=yaml.FullLoader)
        return value;


class TestApi:

    @pytest.mark.parametrize('caseinfo', read_testcase_yaml('get_token.yml'))
    def test_api2(self,caseinfo):
        print(caseinfo)


if __name__ == '__main__':
    pytest.main(['-vs','02-数据驱动.py'])

yaml配置文件的数据
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TMQir92x-1721979295438)(https://i-blog.csdnimg.cn/blog_migrate/5a56413f24221bf918184e2290529282.png)]

如果向设置多条用例的话直接在下面加一条case即可

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-l58yA8dJ-1721979295439)(https://i-blog.csdnimg.cn/blog_migrate/b35f92d798abf6d7ed1bfd5910d9637b.png)]

获取yaml文件数据后输出的结果
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Q7LlmhvX-1721979295440)(https://i-blog.csdnimg.cn/blog_migrate/1bb345fd8a39919a7f91735e7343fc27.png)]