如何使用yaml实现用例数据参数化

这里是用例调用yml的方法

'''导包
 如果没有安装需要安装两个模块pip install pytest
 pip install pyYAML
'''
import yaml
import pytest
# 读取文件
def yaml_data_with_file(file_name):
    with open('./'+ file_name +'.yml',encoding='utf-8') as f:
        return yaml.safe_load(f)
# 读取yaml数据中的key值，这里的yaml文件名是data
def yaml_data_with_key(key):
    return yaml_data_with_file('data')[key]

class Test_01():
    # 单参列表形式等价于
    # @pytest.mark.parametrize('name',['name1','name2'])
    @pytest.mark.parametrize('name',yaml_data_with_key('test_login0'))
    def test_login(self,name):
        print('----')
        print(name)
        print('---')

        #  这里的多参给绑定元组方式
        #  ((value,passd)),[(value,passd),(value1,passd1)]
        #  yaml_data_with_key("value")中的value对应为data.yaml的键值

        # 相当于
    # @pytest.mark.parametrize(('name','password'),[('name3','password3'),('name4','password4')])
    @pytest.mark.parametrize(('name','password'),yaml_data_with_key('test_login1'))
    def test_login2(self,name,password):
        print('-----')
        print(name)
        print(password)
        print('-----')
        # 字典形式的数据
        # 字典形式就相当于{'name':'password'},此时他是一个整体那么在选择时就相当于
    # @pytest.mark.parametrize('dict1',[{'name':'name7','password':'password7'},{'name':'name8','password':'password8'}])
        # 这时就需要抓字典的key来实现参数化
    @pytest.mark.parametrize('dict1',yaml_data_with_key('test_login2'))
    def test_login3(self,dict1):
        print('-----')
        name=dict1['name']
        password=dict1['password']
        print(name)
        print(password)
        print('-----')
# if __name__ == '__main__':
#     pytest.main(['-s','demo.py'])

这下面是data.yml文件的内容

#['name1','name2','name3'] 单参列表方式
test_login0 :
  - 'name1'
  - 'name2'
  - 'name3'
#[('name4','password4'),('name5','password5'),('name6','password6')]多参列表套元组
test_login1 :
  -
    - 'name4'
    - 'password4'
  -
    - 'name5'
    - 'password5'
  -
    - 'name6'
    - 'password6'
#[{'name7':'password7'},{'name8':'password8'},{'name9':'password9'}]
test_login2 :
  -
    name : 'name7'
    password : 'password7'
  -
    name : 'name8'
    password : 'password8'
  -
    name : 'name9'
    password : 'password9'

pytest.ini的文件内容

[pytest]
addopts= -s
testpaths=./
python_files=demo.py
python_classes=Test_*
python_functions=test_*

这三个文件就可以构成一个yaml参数化的简单示例