import json
import pytest


def get_data_json():
    '''
    @return: 返回列表数据
    '''
    with open("test.json") as f:
        lst_data = []
        my_data = json.load(f)
        lst_json = my_data['data']
        for one in lst_json:
            tmp_list = []
            for i in one.keys():  # 遍历one字典中的键值对
                tmp_list.append(one[i])  # 获取键对应的值添加到列表中
            lst_data.append(tmp_list)
        return lst_data


@pytest.mark.parametrize('nam, age, six', get_data_json())
def test_json(nam, age, six):
    print(nam, age, six)


# test.json 文件内容如下：
# {
#  "data": [
#     {
#        "name": "tom",
#        "age": 18,
#        "six": "boy"
#     },
#     {
#        "name": "kike",
#        "age": 19,
#        "six": "girl"
#     },
#     {
#        "name": "rose",
#        "age": 20,
#        "six": "boy"
#     }
#  ]
# }

if __name__ == '__main__':
    # print(get_data_json1())
    pytest.main(['-vs', 'test_json.py'])