如何用pytest驱动yml文件的数据

可以使用pytest-yaml插件来实现这个功能。这个插件可以让pytest直接读取yml文件中的数据，然后将其传递给测试函数。使用这个插件的步骤如下：

1. 安装pytest-yaml插件：

pip install pytest-yaml

2. 在测试文件中导入pytest和pytest-yaml：

import pytest
from pytest_yaml import yaml

3. 创建一个包含测试数据的yml文件，例如：

- name: test_case_1
  input: 1
  expected_output: 2

- name: test_case_2
  input: 2
  expected_output: 4

4. 编写测试函数，并在函数参数中使用pytest的fixture机制来读取yml文件中的数据：

@pytest.fixture(params=yaml("test_data.yml"))
def test_data(request):
    return request.param

def test_function(test_data):
    input_data = test_data["input"]
    expected_output = test_data["expected_output"]
    assert my_function(input_data) == expected_output

在这个例子中，pytest会自动读取test_data.yml文件中的数据，并将其作为参数传递给test_data fixture。然后，测试函数可以从test_data fixture中获取测试数据，并进行测试。这个例子中，测试函数会对my_function函数进行测试，验证其输入和输出是否符合预期。