pytest 学习笔记#

[toc]

pytest是什么#

pytest是python的一种单元测试框架

##pytest安装

1
pip install pytest

pytest使用#

pytest测试样例非常简单，只需要按照下面的规则：

测试文件以test_开头（以_test结尾也可以-建议是test开头）测试类以Test开头，并且不能带有__init__方法测试函数以test_开头断言使用基本的assert即可 fixture的文件名必须是conftest.py

简单例子#

例子1: demo.py

1
def add(a, b):
2
    return a + b

test_demo.py

1
from demo import add
2

3
def test_add():
4
    assert add(1, 2) == 3
5
    assert add(1, 0) == 1
6
    assert add(1, -1) == 0

执行下面命令进行测试：

1
 pytest test_demo.py

例子2: 使用类来组成多个用例的

1
import pytest
2

3
# content of test_class.py
4
class TestClass:
5
    def test_one(self):
6
        x = "this"
7
        assert 'h' in x
8

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

常用的pytest第三方插件#

插件安装方式pip install 插件名称

pytest-sugar#

这个插件是显示执行的进度条

pytest-assume#

这个插件功能是一次性执行完所有测试用例，

不会因为某个测试用例失败而终止后面的用例

test_demo.py

1
import pytest
2
from demo import add
3

4

5
def test_add():
6
    pytest.assume(add(1, 2) == 3)
7
    pytest.assume(add(1, 0) == 2)
8
    pytest.assume(add(1, -1) == 0)

pytest-ordering#

设置测试用例执行顺序

pytest参数化#

1.通过@pytest.mark.parametrize装饰器传参

test_demo.py

1
import pytest
2
from demo import add
3

4
# 参数
5
@pytest.mark.parametrize('a, b, re', [
6
    (3, 4, 7),
7
    (1, -1, 0),
8
    (1, 1.2, 2.2),
9
])
10
# 测试加法
11
def test_add(a, b, re):
12
    pytest.assume(add(a, b) == re)

2.通过读取文件的形式参数化

test.json:

1
[[3, 4, 7], [1, -1, 0], [1, 1.2, 2.2]]

test_demo.py

1
import json
2
import pytest
3
from demo import add
4

5
# 读取json文件
6
def load_json():
7
    with open('test.json', 'r') as f:
8
        re = json.load(f)
9
        return re
10

11
# 把读到文件的数据通过参数传递
12
@pytest.mark.parametrize('a, b, re', load_json())
13
# 测试加法运算
14
def test_add(a, b, re):
15
    pytest.assume(add(a, b) == re)

备注：

1.pytest.mark.parametrize的第一个参数代表测试方法接收的参数个数，第二个参数为接收的数据内容。

2.如果是通过读文件的形式传参需要注意读出来的文件内容是否和要传的参数类型一致。

## pytest执行级别

setup_module 模块级别#

如果在单个模块中有多个测试函数和测试类，则可以选择实现以下方法（只会执行一次）

test_demo.py

1
def setup_module():
2
    """ 模块执行前准备工作 """
3

4
class Test1():
5
    """ 测试类1 """
6

7
class Test2():
8
    """ 测试类2 """
9

10
def teardown_module():
11
    """ 模块执行后的结束工作 """

setup_class 类级别#

在调用类的所有测试方法之前和之后调用以下方法（只会执行一次）

1
class TestDemo():
2
    @classmethod
3
    def setup_class(cls):
4
        """ 类方法执行前准备工作 """
5

6
    def test_add(self):
7
        """ 测试方法 """
8

9
    @classmethod
10
    def teardown_class(cls):
11
        """ 类方法执行后结束工作 """

setup_function 函数级别#

在执行每个函数执行，调用以下方法（只会执行一次）

1
def setup_function():
2
    print("setup_function：每个用例开始前都会执行")
3

4

5
def teardown_function():
6
    print("teardown_function：每个用例结束后都会执行")
7

8

9
def test_one():
10
    print("正在执行----test_one")
11
    x = "this"
12
    assert 'h' in x
13

14

15
def test_two():
16
    print("正在执行----test_two")
17
    x = "hello"
18
    assert hasattr(x, 'check')
19

20

21
def test_three():
22
    print("正在执行----test_three")
23
    a = "hello"
24
    b = "hello world"
25
    assert a in b

fixture#

fixture函数作用#

完成setup和teardown操作，处理数据库、文件等资源的打开和关闭

完成大部分测试用例需要完成的通用操作，例如login、设置config参数、环境变量等

准备测试数据，将数据提前写入到数据库，或者通过params返回给test用例等

有独立的命名，可以按照测试的用途来激活，比如用于functions/modules/class/session

fixture 参数#

scope参数 scope=function：每个test都运行，默认是function的scope scope=class：每个class的所有test只运行一次 scope=module：每个module的所有test只运行一次 scope=session：每个session只运行一次

函数级别#

test_demo.py

1
import pytest
2

3
# 定义fixture函数
4
@pytest.fixture()
5
def fixture_func():
6
    print("\n这是一个fixture函数,用来完成一些提前准备")
7

8
def test_1(fixture_func):
9
    print("测试用例1")
10
    assert 1 == 1
11

12
def test_2(fixture_func):
13
    print("测试用例2")
14
    assert 2 == 2

使用pytest -s test_demo.py 运行，才可以打印出print语句

模块级别#

1
import pytest
2

3

4
@pytest.fixture(scope="module")
5
def fixture_func():
6
    print("\n这是一个fixture函数,用来完成一些提前准备")
7

8
def test_1(fixture_func):
9
    print("测试用例1")
10
    assert 1 == 1
11

12
def test_2(fixture_func):
13
    print("测试用例2")
14
    assert 2 == 2

####session级别fixture

1.先创建一个conftest.py文件，输入以下代码

说明：当有测试用例调用pytest.fixture函数时，

pytest会自动去找conftest.py文件里找pytest.fixture函数，不需要import。

1
import pytest
2
@pytest.fixture(scope="session")
3
def fixture_func():
4
    print("\n这是一个fixture函数,用来完成一些提前准备")

2.创建第一个测试用例文件 test_demo.py

1
def test_1(fixture_func):
2
    print('测试用例1')
3
    assert 1 == 1
4

5
def test_2(fixture_func):
6
    print('测试用例2')
7
    assert 2 == 2

3.再创建一个测试文件test_demo2.py

1
def test_1(fixture_func):
2
    print('测试用例3')
3
    assert 3 == 3
4

5
def test_2(fixture_func):
6
    print('测试用例4')
7
    assert 4 == 4

执行命令：

1
pytest -s test_*

总结：

装饰器@pytest.fixture用于声明一个方法是fixture方法，如果测试用例的参数列表中包含fixture对象，那么测试用例运行之前会先调用fixture方法。

fixture可以指定其方法范围，由参数scope决定。

@pytest.fixture(scope=“function”),function级别的fixture方法在每一个测试用例运行前被调用，待测试用例结束之后再销毁。

如果不给定scope参数，默认情况下就是“function”。

@pytest.fixture(scope=“session”),session级别的fixture方法在每一次会话中只运行一次，也就是在所有测试用例之前运行一次，且被所有测试用例共享。

conftest.py可以认为是pytest中的配置文件，单独管理一些预置的操作，与fixture方法配合，pytest在运行测试用例之前会事先调用conftest.py中预置的fixture方法，然后供所有测试用例使用。

webtest#

1
from webtest import TestApp
2

3

4
def application(environ, start_response):
5
    body = b"hello"
6
    headers = [('Content-Type', 'text/html; charset=utf8'),
7
               ('Content-Length', str(len(body)))]
8
    start_response('200 OK', headers)
9
    return [body]
10

11

12
def test_qiniu():
13
    app = TestApp(application)
14
    resp = app.get("/")
15
    assert resp.status == '200 OK'

pytest 学习笔记#

pytest是什么#

pytest使用#

简单例子#

常用的pytest第三方插件#

pytest-sugar#

pytest-assume#

pytest-ordering#

pytest参数化#

setup_module 模块级别#

setup_class 类级别#

setup_function 函数级别#

fixture#

fixture函数作用#

fixture 参数#

函数级别#

模块级别#

webtest#

相关阅读