pytest 数据驱动怎么写

在 Pytest 中使用数据驱动的方式可以通过使用装饰器 `@pytest.mark.parametrize` 来实现。这个装饰器可以帮助我们传递多组不同的参数值给测试函数。

下面是一个示例，演示了如何使用 `@pytest.mark.parametrize` 来进行数据驱动测试：

import pytest

@pytest.mark.parametrize("input, expected", [
    (4, 2),
    (9, 3),
    (16, 4)
])
def test_square_root(input, expected):
    assert input ** 0.5 == expected

在这个例子中，`test_square_root` 函数被装饰器 `@pytest.mark.parametrize` 所修饰。装饰器的第一个参数是一个字符串，用于指定测试函数的参数名称。接下来是一个由多个元组组成的列表，每个元组包含一组输入和对应的预期输出。

在运行测试时，`@pytest.mark.parametrize` 会为每组参数值生成一个独立的测试用例，并将这些参数传递给测试函数。在上面的示例中，`test_square_root` 函数会被调用三次，分别传入参数 `(4, 2)`、`(9, 3)` 和 `(16, 4)`。

这样，你就可以通过数据驱动的方式来编写更加灵活和可扩展的测试用例了。

相关问题

pytest数据驱动

pytest的数据驱动有两种方式，一种是使用parametrize驱动，另一种是使用yaml驱动。使用parametrize驱动可以通过pytest自带的pytest.mark.parametrize()装饰器来实现。这种方式可以在测试用例中传入多组参数，从而减少重复的测试代码。另外，还可以使用pytest.fixture装饰器进行测试的setup、teardown以及fixture共享的测试。另一种方式是使用yaml驱动，通过在测试用例所在的文件中使用yaml格式的数据文件，将测试数据和预期结果存储在同一个地方，方便管理和维护。这种方式使得测试用例的数据和代码分离，提高了测试用例的可读性和可维护性。通过这两种数据驱动的方式，我们可以更加灵活地进行测试，并且可以减少冗余的代码，提高测试效率。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [自动化测试框架Pytest（四） ——数据驱动](https://blog.csdn.net/m0_58026506/article/details/126274826)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
- *2* *3* [Pytest 测试框架——数据驱动](https://blog.csdn.net/weixin_41754309/article/details/113185895)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]

pytest 数据驱动

pytest的数据驱动有两种方式，分别是parametrize驱动和yaml驱动。parametrize驱动是通过使用@pytest.mark.parametrize装饰器将测试用例参数化，可以在测试用例中传入多组参数进行测试。这种方式适合参数较少且比较简单的情况。而yaml驱动是通过读取yaml文件中的数据作为测试用例的参数，可以灵活地组织和管理测试数据。这种方式适合参数较多或者参数需要嵌套结构的情况。

当使用parametrize驱动时，可以通过在装饰器中指定参数名和参数值的方式传入参数，支持传入列表、元组或者字典等多种数据结构。这样可以方便地进行多组参数的测试。例如，可以使用@pytest.mark.parametrize("name,age", [['百里',11],['婉儿',22],['露露',33]])来传入多个姓名和年龄的参数。

当使用yaml驱动时，需要先引入yaml_util模块中的read_yaml函数，并通过该函数读取yaml文件中的数据。读取后的数据可以直接作为参数传入测试用例中。这样可以将测试数据和测试逻辑分离，提高代码的可维护性和可读性。

综上所述，pytest的数据驱动可以通过parametrize和yaml两种方式实现，分别适用于不同的测试场景和需求。