如何使用yaml

import pytest
import yaml

def func(x):
    return x + 1

@pytest.mark.parametrize(["a","b"],yaml.safe_load(open("./data.yaml")))
def test_answer(a,b):
    assert func(a) == b