写一个简单的pytest框架
时间: 2023-08-02 16:07:53 浏览: 40
下面是一个简单的pytest框架的示例:
```
# test_sample.py
def test_add():
assert (1 + 2) == 3
def test_subtract():
assert (4 - 2) == 2
```
在这个示例中,我们定义了两个测试用例:test_add和test_subtract。这两个测试用例分别测试了加法和减法的功能是否正确。在每个测试用例中,我们使用assert语句来断言测试结果是否符合预期。
现在,我们可以使用pytest命令来运行这个测试用例:
```
$ pytest test_sample.py
```
运行结果如下:
```
============================= test session starts ==============================
platform darwin -- Python 3.7.6, pytest-6.2.2, py-1.10.0, pluggy-0.13.1
rootdir: /Users/username/pytest_demo
collected 2 items
test_sample.py .. [100%]
============================== 2 passed in 0.01s ===============================
```
可以看到,这个测试用例运行成功,两个测试用例都通过了测试。
当然,这只是一个最简单的pytest框架示例,实际使用中,我们可以使用更多的pytest特性,如参数化测试、fixture等,让测试更加灵活和高效。