9.1. pytest介绍

9.1.1. pytest简介

pytest是python的一种单元测试框架,与python自带的unittest测试框架类似,但是比unittest框架使用起来更简洁,效率更高。根据pytest的官方网站介绍,它具有如下特点:

  • 非常容易上手,入门简单,文档丰富,文档中有很多实例可以参考

  • 能够支持简单的单元测试和复杂的功能测试

  • 支持参数化

  • 执行测试过程中可以将某些测试跳过,或者对某些预期失败的case标记成失败

  • 支持重复执行失败的case

  • 支持运行由nose, unittest编写的测试case

  • 具有很多第三方插件,并且可以自定义扩展

  • 方便持续集成

9.1.2. 安装pytest

与安装其他的python类库无异,直接使用pip安装。

pip install -U pytest
pip install -U pytest-html
pip install -U pytest-rerunfailures

安装完成后,可以验证安装的版本:

py.test --version

9.1.3. 一个实例

我们可以通过下面的实例,看看使用py.test进行测试是多么简单。

# content of test_sample.py

def func(x):
    return x+1

def test_func():
    assert func(3) == 5

这里我们定义了一个被测试函数func,该函数将传递进来的参数加1后返回。我们还定义了一个测试函数test_func用来对func进行测试。test_func中我们使用基本的断言语句assert来对结果进行验证。 下面来运行这个测试:

$ py.test
=========================== test session starts ============================
platform linux -- Python 3.4.1 -- py-1.4.27 -- pytest-2.7.1
rootdir: /tmp/doc-exec-101, inifile:
collected 1 items
test_sample.py F
================================= FAILURES =================================
_______________________________ test_answer ________________________________
def test_answer():
> assert func(3) == 5
E assert 4 == 5
E + where 4 = func(3)
test_sample.py:5: AssertionError
========================= 1 failed in 0.01 seconds =========================

执行测试的时候,我们只需要在测试文件test_sample所在的目录下,运行py.test即可。pytest会在当前的目录下,寻找以test开头的文件(即测试文件),找到测试文件之后,进入到测试文件中寻找test_开头的测试函数并执行。

通过上面的测试输出,我们可以看到该测试过程中,一个收集到了一个测试函数,测试结果是失败的(标记为F),并且在FAILURES部分输出了详细的错误信息,帮助我们分析测试原因,我们可以看到”assert func(3) == 5”这条语句出错了,错误的原因是func(3)=4,然后我们断言func(3) 等于 5。

9.1.4. 再一个实例

当需要编写多个测试样例的时候,我们可以将其放到一个测试类当中,如:

# content of test_class.py

class TestClass:
    def test_one(self):
        x = "this"
        assert 'h' in x

    def test_two(self):
        x = "hello"
        assert hasattr(x, 'check')

我们可以通过执行测试文件的方法,执行上面的测试:

$ py.test -q test_class.py
.F
================================= FAILURES =================================
____________________________ TestClass.test_two ____________________________
self = <test_class.TestClass object at 0x7fbf54cf5668>
def test_two(self):
x = "hello"
> assert hasattr(x, 'check')
E assert hasattr('hello', 'check')
test_class.py:8: AssertionError
1 failed, 1 passed in 0.01 seconds

从测试结果中可以看到,该测试共执行了两个测试样例,一个失败一个成功。同样,我们也看到失败样例的详细信息,和执行过程中的中间结果。pytest中用点号(.)表示一条用例被执行并通过,用F来标识一条用例被执行并失败,如果想查看详情可以在py.test命令后面加上-v或者–verbose选项,具体效果可以自行尝试。

9.1.5. 如何获取帮助信息

py.test --version # shows where pytest was imported from
py.test --fixtures # show available builtin function arguments
py.test -h | --help # show help on command line and config file options

9.1.6. 使用raises可以帮助我们断言某些代码会引发某个异常

# content of test_sysexit.py
import pytest
def f():
    raise SystemExit(1)

def test_mytest():
    with pytest.raises(SystemExit):
        f()

9.1.7. 多个测试的类

当我们开发了多个测试时,可能会把它们分组到一个类中,我们现在可以使用pytest创建一个包含多个测试的类

# content of test_class.py
class TestClass(object):
    def test_one(self):
        x = "this"
        assert 'h' in x

    def test_two(self):
        x = "hello"
        assert hasattr(x, 'check')

pytest会发现所有test_命名的函数,没有必要继承任何东西,我们可以简单地通过传递它的文件名来运行测试:

$ pytest -q test_class.py
.F                                                                   [100%]
================================= FAILURES =================================
____________________________ TestClass.test_two ____________________________

self = <test_class.TestClass object at 0xdeadbeef>

    def test_two(self):
        x = "hello"
>       assert hasattr(x, 'check')
E       AssertionError: assert False
E        +  where False = hasattr('hello', 'check')

test_class.py:8: AssertionError
1 failed, 1 passed in 0.12 seconds

9.1.8. 使用内置fixture

fixture是pytest中的一个特性,fixture可以请求任意资源,用文字不太好理解,我们就通过实例来理解吧。首先,通过以下命令可以找出所有pytest内置的fixture:

$ pytest --fixtures

我们就以tmpdir这个内置的fixture来演示,tmpdir能返回一个唯一的临时目录路径,新建一个test_tmpdir.py文件,输入以下代码:

def test_needsfiles(tmpdir):
    print (tmpdir)
    assert 0

在测试函数的参数中列出tmpdir,pytest将在执行测试函数之前查找并调用fixture工厂来创建资源:

$ pytest -q test_tmpdir.py

在测试运行之前,pytest会创建一个唯一的,供每个测试调用的临时目录

=================================== FAILURES ===================================
_________________________ TestBuilding.test_needsfiles _________________________

self = <test_suites.test_building.test_building_process.TestBuilding object at 0x105ed2ba8>
tmpdir = local('/private/var/folders/08/lxy0ywy90mj9ck0rz1tq0y_r0000gn/T/pytest-of-lvjunjie/pytest-2/test_needsfiles0')

    def test_needsfiles(self, tmpdir):
        print(tmpdir)
>       assert 0
E       assert 0

test_building_process.py:102: AssertionError
----------------------------- Captured stdout call -----------------------------
/private/var/folders/08/lxy0ywy90mj9ck0rz1tq0y_r0000gn/T/pytest-of-lvjunjie/pytest-2/test_needsfiles0
=========================== 1 failed in 0.12 seconds ===========================
Process finished with exit code 0

9.1.9. fixture的scope参数

scope参数有四种,’function’,’module’,’class’,’session’,默认为function。

  • function:每个test都运行,默认是function的scope

  • class:每个class的所有test只运行一次

  • module:每个module的所有test只运行一次

  • session:每个session只运行一次

setup和teardown操作

  • setup,在测试函数或类之前执行,完成准备工作,例如数据库链接、测试数据、打开文件等

  • teardown,在测试函数或类之后执行,完成收尾工作,例如断开数据库链接、回收内存资源等

  • 备注:也可以通过在fixture函数中通过yield实现setup和teardown功能

9.1.10. 最佳实践

其实对于测试而言,特别是在持续集成环境中,我们的所有测试最好是在虚拟环境中。这样不同的虚拟环境中的测试不会相互干扰的。 由于我们的实际工作中,在同一个Jenkins中,运行了好多种不同项目册的测试,因此,各个测试项目运行在各自的虚拟环境中。

将pytest安装在虚拟环境中:

virtualenv .        # create a virtualenv directory in the current directory
source bin/activate # on unix