写一个pytest执行的测试用例+在run.py中调用的方法
时间: 2024-12-08 21:22:00 浏览: 22
在Python中,Pytest是一个流行的测试框架,我们可以编写一个简单的测试用例并将其放在一个文件中,然后在`run.py`或其他合适的主入口文件中调用它。这里是一个基本的例子:
首先,在`test_case/test_py_1.py`文件中创建一个测试用例,假设我们有一个名为`addition`的方法在`math_module.py`里:
```python
# test_case/test_py_1.py
import math_module # 假设存在这样的模块,包含一个addition函数
def test_addition():
assert math_module.addition(1, 2) == 3, "Addition of 1 and 2 should be 3"
assert math_module.addition(0, 0) == 0, "Zero addition should return zero"
```
在这个例子中,我们定义了一个测试函数`test_addition`,它通过`assert`断言`addition`函数的行为符合预期。
接着,在`run.py`或项目的主要入口文件中,你可以导入并运行测试模块:
```python
# run.py 或 main.py
import pytest
import os
# 如果项目位于其他地方,可以加上os.getcwd() 获取当前工作目录
if __name__ == "__main__":
current_dir = os.path.dirname(__file__)
test_dir = os.path.join(current_dir, "test_case")
pytest.main([test_dir])
```
上述代码中,我们导入了pytest库,然后获取了测试文件夹的路径,最后调用`pytest.main()`来执行所有的测试。
注意,你需要确保`math_module`已经定义了`addition`函数,或者修改测试用例使其指向实际存在的函数。
阅读全文