Pytest 结合数据驱动-json

 

 

案例：

从json文件中取值判断是否第一二行的值之和等于第三行的值

1.创建对应目录，准备好对应json文件

 

 

  2.创建operation.py

def my_add(x,y):
    result = x + y
    return result

3.创建test_add.py

关键代码：get_json()

注意json loads出来的是字典，把字典的values取出来并转型到list

def get_json():
    with open('../data/params.json','r',encoding='utf-8') as f:
        data = json.loads(f.read())
        data = list(data.values())
        return data

class TestWithCSV:
    @pytest.mark.parametrize('x,y,expected', get_json())
    def test_add(self, x, y, expected):
        assert my_add(int(x), int(y)) == int(expected)