💎一站式轻松地调用各大LLM模型接口,支持GPT4、智谱、星火、月之暗面及文生图 广告
[TOC] ## pytest 介绍 pytest是python的一种单元测试框架,与python自带的unittest测试框架类似,但是比unittest框架使用起来更简洁,效率更高。根据pytest的官方网站介绍,它具有如下特点: * 非常容易上手,入门简单,文档丰富,文档中有很多实例可以参考 * 能够支持简单的单元测试和复杂的功能测试 * 支持参数化 * 执行测试过程中可以将某些测试跳过,或者对某些预期失败的case标记成失败 * 支持重复执行失败的case * 支持运行由nose, unittest编写的测试case * 具有很多第三方插件,并且可以自定义扩展 * 方便的和持续集成工具集成 ## pytest 安装 ```cmd pip install -U pytest ``` ## pytest 的使用 ### Pytest测试样例的命名规则 1. 测试文件以test_开头或结尾(否则用py.test命令行不能自动识别) 2. 测试类以Test开头,且不能带有init方法 3. 测试函数以test_开头 4. 断言使用assert 5. fixture的文件名必须是`conftest.py` ### 创建一个最简单的例子 test_sample.py ```python def func(x): return x + 1 def test_answer_right(): assert func(3) == 4 def test_answer_wrong(): assert func(3) == 5 ``` 执行 ```cmd pytest test_sample.py ``` 运行结果如下: ```cmd ============================= test session starts ============================= platform win32 -- Python 3.5.2, pytest-3.6.3, py-1.5.4, pluggy-0.6.0 rootdir: E:\workspace\python_leaning, inifile: collected 2 items test_sample.py .F test_sample.py:12 (test_answer_wrong) def test_answer_wrong(): > assert func(3) == 5 E assert 4 == 5 E + where 4 = func(3) test_sample.py:14: AssertionError [100%] ================================== FAILURES =================================== ______________________________ test_answer_wrong ______________________________ def test_answer_wrong(): > assert func(3) == 5 E assert 4 == 5 E + where 4 = func(3) test_sample.py:14: AssertionError ===================== 1 failed, 1 passed in 0.09 seconds ====================== ``` 可以发现,执行了两个用例,一个成功,一个失败。 上述命令可以使pytest自动查找 test_sample.py 文件下所有的格式为:`test_*`的方法,然后执行用例.执行顺序为方法的顺序. 如果有多个py文件,可以直接使用下面命令 ``` pytest ``` 这个命令会找出当前目录下所有格式为`test_*.py `或 `*_test.py` 的文件 然后执行用例(文件内格式为`test_*`的方法) ### 在类中分组组织用例 test_sample.py ```python import pytest class TestSample(object): @pytest.fixture() def count(self): print('init count') return 10 def test_answer_1(self, count): print('test_answer_1 get count: %s' % count) assert count == 10 def test_answer_2(self, count): print('test_answer_2 get count: %s' % count) assert count == 10 ``` 运行命令 ``` pytest -s -q ``` `-s`: 是为了输出打印日志,如果不输入-s,是没有日志输出的。 `-q`:quiet 报道模式,这样输出会好看点 运行结果如 ```cmd init count test_answer_1 get count: 10 .init count test_answer_2 get count: 10 . 2 passed in 0.05 seconds ``` ## fixtures 说到测试框架自然要说到setup和teardown两个方法。 * setup:是用来做准备操作.一般用来初始化资源. * teardown:是用来做收尾操作.一般用了关闭资源. pytest的setup和teardown是利用`@pytest.fixture`这个注释来完成的.不仅可以完成初始化操作,初始化后如果有数据需要传给用例使用也是非常方便! 通过`@pytest.fixture() `注释会在执行测试用例之前初始化操作.然后直接在测试用例的方法中就可以拿到初始化返回的参数(<span style="color:red;">参数名</span>要和<span style="color:red;">初始化的方法名</span>**一样**) 在这个实例中,初始化的方法名为`count`,初始化后返回 10。当在测试用例方法中传入与初始化方法名同为`count`的参数时,相对于传入参数`count=10` ### 在fixtures中控制setup与teardown test_sample.py: ```pyhon import pytest class TestSample(object): @pytest.fixture() def count(self,request): print("init count") def fin(): print("teardown count") request.addfinalizer(fin) # teardown return 10 # provide the fixture value def test_answer_1(self, count): print('test_answer_1 get count: %s' % count) assert count == 10 def test_answer_2(self, count): print('test_answer_2 get count: %s' % count) assert count == 10 ``` 运行命令 ``` pytest -s -q``` ```cmd init count test_answer_1 get count: 10 .teardown count init count test_answer_2 get count: 10 .teardown count ``` 在fixture方法中, request.addfinalizer(),会将某函数作为teardown,在用例执行之后执行。 ### fixture 的参数scope `fixture` 有一个域(`scope`)的概念,用来指定该 `fixture` 的使用范围. 它有五种作用域,如`session`/`module`/`class`/`function`(default). - function:每个测试函数之前执行一次 - class:每个测试类之前执行一次 - module:每个module之前执行一次 - session:每次session之前执行一次,即每次测试执行一次 观察上面例子的运行结果,可以发现`init count`与`teardown count` 都打印了两次,那是因为`fixture`默认域是`function`,即每执行一个用例方法前,都会执行初始化方法一次。 为了演示`fixture`的作用域,这里修改一下工程的目录结构,将`fixture`放到一个名称为`conftest.py`(这是pytest规定的命名)的文件中. >[info] 如果一个fixture需要共享,不用在每个py文件中写一遍,写在conftest.py文件中就好。 `conftest.py` ```python import pytest @pytest.fixture(scope="session") def count(): print("init count") yield 10 print("teardown count") ``` `test_sample.py`: ```python class TestSample(object): def test_answer_1(self, count): print('test_answer_1 get count: %s' % count) assert count == 10 def test_answer_2(self, count): print('test_answer_2 get count: %s' % count) assert count == 10 ``` 运行命令 ``` pytest -s -q``` ```cmd init count test_answer_1 get count: 10 .test_answer_2 get count: 10 .teardown count 2 passed in 0.06 seconds ``` 这是在使用 seesion 域的情况下,可以发现`init count` 与 `teardown` 只执行一次.假如我们需要这种情况,每一个类都执行一次初始化方法,那么很简单直接把域改成 class 即可,同理 module 是基于模块的. ### fixture 的参数params 在`test_sample.py`中添加一个带params的fixture,并且在用例中调用这个fixture ```python import pytest @pytest.fixture(params=[1, 2, 3]) def test_data(request): return request.param class TestSample(object): @pytest.mark.skip() def test_answer_1(self, count): print('test_answer_1 get count: %s' % count) assert count == 10 @pytest.mark.skip() def test_answer_2(self, count): print('test_answer_2 get count: %s' % count) assert count == 10 def test_answer_3(self, test_data): print('test_data: %s' % test_data) assert test_data in [1, 2, 3] ``` 执行命令```pytest -s -q``` ```cmd test_data: 1 .test_data: 2 .test_data: 3 . 3 passed, 2 skipped in 0.05 seconds ``` 这里通过`@pytest.mark.skip()`跳过了一些不想要执行的用例 通过执行结果可知,fixture还可以带参数,把参数赋值给params,默认是None。对于param里面的每个值,fixture都会去调用执行一次,就像执行for循环一样把params里的值遍历一次。 ## 用例运行顺序 pytest-ordering pytest按照在.py中的出现位置(`从上到下`)执行用例,可以通过`pytest-ordering` 插件来改变执行顺序 ### 安装pytest-ordering ```cmd pip install pytest-ordering ``` ### 在用例前,使用装饰器 `@pytest.mark.run` 指定用例顺序 当用例中存在 `@pytest.mark.run` 时,先按照order从小到大执行,然后再从上往下执行。 ```python import pytest @pytest.fixture(params=[1, ]) def test_data(request): return request.param class TestSample(object): @pytest.mark.run(order=2) def test_answer_1(self, count): print('test_answer_1 get count: %s' % count) assert count == 10 @pytest.mark.run(order=1) def test_answer_2(self, count): print('test_answer_2 get count: %s' % count) assert count == 10 def test_answer_3(self, test_data): print('test_data: %s' % test_data) assert test_data in [1, 2, 3] ``` 执行命令 ```pytest -s -q``` ```cmd init count test_answer_2 get count: 10 .test_answer_1 get count: 10 .test_data: 1 .teardown count 3 passed in 0.09 seconds ``` ## pytest生成html测试报告 pytest-html pytest-html是py.test的一个插件,它为测试结果生成一个HTML报告。 ### 安装pytest-html ```cmd pip install pytest-html ``` ### 运行测试用例 ```cmd pytest --html=report.html ``` 运行后,在当前目录下生产`report.html`,如下 ![](https://box.kancloud.cn/7cae74e0235dfe2869b9a7806c330fc4_1388x565.jpg) ## 在代码中执行用例 有些时候需要在代码中调用执行case的方法.而不是通过command的方式.我们可以在python代码中加入下面代码即可: ```python args = ['-s', '-q'] pytest.main(args) ``` ## 参考文档 作为pytest新手入门,先快速掌握如上部分内容,更多有用的pytest内容,后续会继续更新。 官方文档: [https://docs.pytest.org/en/latest/contents.html](https://docs.pytest.org/en/latest/contents.html) <hr style="margin-top:100px"> :-: ![](https://box.kancloud.cn/2ff0bc02ec938fef8b6dd7b7f16ee11d_258x258.jpg) ***微信扫一扫,关注“python测试开发圈”,了解更多测试教程!***