前言
pytest 是 python 的一个第三方单元测试框架，它继承自 python 自带的单元测试框架unittest，兼容 unittest。
相比unittest，pytest的可扩展性更高，也是目前最为流行的 python 单元测试框架。至于它扩展性表现在哪些方面，我们需在后续的学习中一点一点进行总结。
简单使用
1 安装
安装pytest
pip install -U pytest
验证安装是否成功
pytest --version
2 用例编写
为了先让大家对pytest的使用有个粗浅的认识，接下来我们使用pytest对两个自定义接口进行简单测试，其中查询所有用户信息接口为get请求，注册用户接口为post请求，编写脚本test_demo.py，代码如下：
import pytest import requests, json class TestDemo: def test_get_all_users(self): '''查询所有用户信息''' url = "http://127.0.0.1:5000/users" # 请求接口 res = requests.get(url=url).text res = json.loads(res) print(res) # 断言 assert res['code'] == 0 def test_register(self): '''注册用户''' headers = {"Content-Type": "application/json;charset=utf8"} url = "http://127.0.0.1:5000/register" data = { "username": "张学友", "password": "123456", "sex": "0", "telephone": "13823456789", "address": "北京东城区" } # 请求接口 res = requests.post(url=url, headers=headers, json=data).text res = json.loads(res) print(res) # 断言 assert res['code'] == 0 if __name__ == '__main__': pytest.main()
注意，测试类中不能定义__init__方法。
3 用例执行
- 方式一：在python代码里调用pytest，使用
pytest.mian()
# 执行当前目录下的测试用例 pytest.main() # 执行指定的测试用例 pytest.main("testcase/test_one.py") # 加参数如-s，执行参数根据需要进行添加，后面文章会对常用的参数进行说明 pytest.main(["-s", "testcase/test_one.py"])
- pycharm控制台输出如下：
- 方式二：命令行执行
pytest命令行执行pytest + 测试文件路径，示例如下
pytest E:/apiAutoTest/test_demo.py # 加参数如-s pytest -s E:/apiAutoTest/test_demo.py
- 除了上面的直接通过pytest命令调用外，还可以通过命令在python解释器里调用，
python -m 测试文件完整路径，如下：
python -m pytest E:/apiAutoTest/test_demo.py
结果如下：
总结
如果使用unittest框架编写上面的测试用例的话，代码应该如下：
import unittest import requests, json class TestDemo(unittest.TestCase): def test_get_all_users(self): '''查询所有用户信息''' url = "http://127.0.0.1:5000/users" res = requests.get(url=url).text res = json.loads(res) self.assertEqual(res['code'], 0) def test_register(self): '''注册用户''' headers = {"Content-Type": "application/json;charset=utf8"} url = "http://127.0.0.1:5000/register" data = { "username": "张学友", "password": "123456", "sex": "0", "telephone": "13823456789", "address": "北京东城区" } res = requests.post(url=url, headers=headers, json=data).text res = json.loads(res) self.assertEqual(res['code'], 0) if __name__ == '__main__': unittest.main()
由这个简单的测试用例，我们可以比较出pytest与unittest 在测试用例的编写及执行时不一样的地方：
pytest框架编写测试用例时，只需要引入 pytest 模块，使用python源生态的 assert 进行断言，且可以使用自带的命令行执行测试用例。- 而
unittest框架编写测试用例时，自定义的测试类需要继承unittest.TestCase，断言则使用的是unittest提供的断言方式（如 assertEqual、assertTrue、assertIn等），没有提供命令行方式执行测试用例。
当然，pytest与unittest的区别远不止这些，待我们后续慢慢道来。
