pytest中yaml文件用法
时间: 2023-10-04 12:05:59 浏览: 79
在pytest中使用yaml文件可以用于存储测试数据、配置文件等信息,方便测试用例的编写和维护。
首先需要安装PyYAML模块,可以使用pip安装:
```
pip install PyYAML
```
假设我们有一个yaml文件`test_data.yaml`,内容如下:
```yaml
- name: test1
input: "hello"
expected_output: "HELLO"
- name: test2
input: "world"
expected_output: "WORLD"
```
那么在测试用例中可以这样使用:
```python
import yaml
def test_uppercase():
with open("test_data.yaml", 'r') as f:
test_data = yaml.safe_load(f)
for data in test_data:
input_str = data['input']
expected_output = data['expected_output']
assert input_str.upper() == expected_output
```
在测试用例中先打开yaml文件并读取其中的数据,然后按照数据格式进行测试即可。
需要注意的是,yaml文件中的数据格式必须按照yaml语法规范来编写,否则会出现解析错误。另外,使用PyYAML读取yaml文件时,需要使用`safe_load`方法来避免安全问题。
阅读全文