pytest单元测试笔记

最新推荐文章于 2025-07-12 15:39:01 发布

原创最新推荐文章于 2025-07-12 15:39:01 发布 · 544 阅读

2 ·

CC 4.0 BY-SA版权

测试专栏收录该内容

9 篇文章

订阅专栏

本文介绍了Python 3.6.8版本下pytest单元测试的使用。包括使用虚拟环境pipenv安装pytest包，创建第一个单元测试。还阐述了pytest fixtures的特点，如显式、模块化、可伸缩，以及在不同范围跨测试共享夹具实例，和更高范围灯具先实例化的规则。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

单元测试–pytest使用

学习链接

官方文档

安装(使用的是python3.6.8版本)

使用虚拟环境 pipenv:先进入程序所在文件夹，终端运行pipenv install,安装虚拟环境
在虚拟环境中安装pytest包，终端运行pipenv install pytest,安装完成
进入虚拟环境pipenv shell,查看包是否安装正确版本：pytest --version

创建第一个单元测试 (设为unit1.py)

def func(x):
		return(x+1)
def test_answer():   # 单元测试前缀test_
		assert func(3) == 5

运行：pipenv run python -m pytest unit1.py
测试结果

pytest fixtures:显式，模块化，可伸缩

测试夹具目的：提供一个固定的基线，测试可以可靠地重复执行。

import pytest
@pytest.fixture
def smtp_connection():
		import smtplib
		return smtplib.SMTP("smtp.gmail.com",587,timeout=5)
def test_echo(smtp_connection):
		response,msg = smtp_connection.ehlo()
		assert response == 250
		assert 0

运行：pipenv run python -m pytest unit1.py
在这里插入图片描述

范围：在类，模块或会话中跨测试共享夹具实例

将fixture函数放入单独的conftest.py文件中，以便来自目录中多个测试模块的测试可以访问fixture函数：

import pytest
import smtplib

@pytest.fixture(scope="module")
def smtp_connection():
    return smtplib.SMTP("smtp.gmail.com", 587, timeout=5)

再次使用夹具的名称，smtp_connection您可以通过smtp_connection在任何测试或夹具功能中（在所在目录中或下面conftest.py）将名称列为输入参数来访问其结果：

def test_ehlo(smtp_connection):
    response, msg = smtp_connection.ehlo()
    assert response == 250
    assert b"smtp.gmail.com" in msg
    assert 0  # for demo purposes

def test_noop(smtp_connection):
    response, msg = smtp_connection.noop()
    assert response == 250
    assert 0  # for demo purposes

在这里插入图片描述
看到相同的（模块范围的）对象被传递到两个测试函数中，因为pytest在回溯中显示传入的参数值。因此，使用两个测试函数的运行速度与单个测试函数一样快，因为它们重用相同的实例。

更高范围的灯具首先被实例化

在功能的功能请求中，较高范围（例如session）的夹具首先比较低范围的夹具（例如function或class）实例化。相同范围的灯具的相对顺序遵循测试功能中声明的顺序，并尊重灯具之间的依赖关系。

import pytest


@pytest.fixture(scope="session")
def s1():
    pass


@pytest.fixture(scope="module")
def m1():
    pass


@pytest.fixture
def f1(tmpdir):
    pass


@pytest.fixture
def f2():
    pass


def test_foo(f1,m1,f2,s1):
    assert 0

请求的灯具test_foo将按以下顺序实例化：

s1：是最高范围的夹具（session）。
m1：是第二个最高范围的夹具（module）。
tmpdir：是一个function-scoped fixture，需要f1：它需要在这一点实例化，因为它是一个依赖项f1。
f1：是参数列表中的第一个function-scoped fixture test_foo。
f2：是参数列表中最后一个function-scoped fixture test_foo。