一: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)]