Python测试框架之—— pytest介绍与示例

Pytest是一个功能强大且易于使用的Python测试框架,它提供了丰富的功能和灵活的用法,使得编写和运行测试变得简单而高效。

一、Pytest的特点

  1. 简单灵活:Pytest的语法简洁清晰,容易上手,并且支持复杂的测试场景。
  2. 自动发现测试:Pytest能够自动发现项目中的测试文件和测试函数,无需手动编写繁琐的配置。
  3. 丰富的断言库:Pytest内置了丰富的断言库,可以方便地进行测试结果的判断。
  4. 支持参数化测试:Pytest支持参数化测试,允许对多组输入进行测试,提高测试效率。
  5. 插件丰富:Pytest拥有丰富的插件生态系统,如pytest-selenium(集成selenium)、pytest-html(生成HTML测试报告)、pytest-rerunfailures(失败case重复执行)、pytest-xdist(多CPU分发)等,可以通过插件扩展各种功能。
  6. 可以与Jenkins集成:Pytest能够很好地与Jenkins等持续集成工具集成,方便自动化测试流程的构建。

二、Pytest的安装

Pytest的安装非常简单,可以使用pip命令进行安装:

pip install -U pytest

安装完成后,可以通过运行pytest --version来验证安装是否成功,如果看到版本号则说明安装成功。

三、Pytest的基本使用

  1. 测试文件命名 :测试文件应该以test_开头或以_test结尾,例如test_login.pyuser_test.py
  2. 测试函数命名 :测试函数应该以test开头,例如def test_login_success():
  3. 测试类命名 :如果使用类来组织测试用例,测试类应该以Test开头,且不能有__init__方法。测试类中的测试函数同样需要以test开头。
  4. 执行测试 :可以通过命令行直接运行pytest命令来执行测试,或者通过编写Python代码调用pytest.main()函数来执行测试。

四、Pytest的运行参数

Pytest提供了丰富的运行参数,以支持不同的测试需求。以下是一些常用的运行参数:

  • -v:打印详细运行日志。
  • -s:打印测试中的print输出。
  • -k:通过关键字表达式过滤要执行的测试用例。
  • -x:运行用例失败后立即停止执行。
  • --maxfail=num:当用例失败数达到某个设定的值时停止运行。
  • -m:运行所有被特定标记装饰的测试用例。

五、Pytest的fixture

Pytest的fixture是一个非常强大的功能,它主要用于测试环境的初始化和清理工作。fixture可以代替传统的setup和teardown函数,提供更加灵活和强大的功能。

fixture通过@pytest.fixture()装饰器进行定义,可以使用scope参数控制fixture的作用范围(如function、class、module、session)。在测试用例中,可以通过函数参数的形式传入fixture,以便在测试前后进行环境的初始化和清理。

六、Pytest的断言

Pytest的断言主要使用Python的原生断言方法,如==inis等。此外,Pytest还提供了assert语句的扩展功能,可以方便地生成断言失败的详细信息。

七、Pytest的skip和xfail

Pytest支持在测试用例中跳过某些不需要执行的测试,或者将某些预期会失败的测试标记为xfail(expected failure)。这有助于在开发过程中管理那些尚未实现或存在已知问题的功能。

八、Pytest的示例

这里的Python项目的结构很简单,如下所示:

project/  
├── src/  
│   ├── module1.py  
│   └── module2.py  
└── tests/  
    ├── test_module1.py  
    └── test_module2.py

这里以实现斐波那契函数及对其进行测试为例:

斐波那契函数是一个有趣的古典数学问题,著名意大利数学家Fibonacci曾提出一个问题:有一对小兔子,从出生后第3个月起每个月都生一对兔子。小兔子长到第3个月后每个月又生一对兔子。按此规律,假设没有兔子死亡,第一个月有一对刚出生的小兔子,问第n个月有多少对兔子?

在src目录下新增 fibonacci.py , 内容如下:

def fibonacci(n):  
    if n <= 0:  
        return 0  
    elif n == 1:  
        return 1  
    a, b = 0, 1  
    for _ in range(2, n + 1):  
        a, b = b, a + b  
    return b  

在tests目录下新增测试文件:test_fibonacci.py,内容如下:

import pytest
import sys  
import os  
  
# 获取当前文件的目录  
current_dir = os.path.dirname(os.path.abspath(__file__))  

# 获取当前文件的父目录路径  
parent_dir = os.path.dirname(current_dir)  

# 假设src目录与当前文件(test_fibonacci.py)在同一级别  
src_dir = os.path.join(parent_dir, 'src')  
# 将src目录添加到sys.path中,以便可以导入其中的模块  
sys.path.insert(0, src_dir)  
  
# 现在可以导入src目录下的fibonacci模块了  
from fibonacci import fibonacci  

def test_fibonacci_zero():  
    assert fibonacci(0) == 0  
  
def test_fibonacci_one():  
    assert fibonacci(1) == 1  
  
def test_fibonacci_two():  
    assert fibonacci(2) == 1  
  
def test_fibonacci_three():  
    assert fibonacci(3) == 2  
  
def test_fibonacci_four():  
    assert fibonacci(4) == 3  
  
def test_fibonacci_large_number():  
    # 选择一个较大的数字进行测试,确保算法在较大输入时也能正确工作  
    # 这里选择了第10个斐波那契数,因为55是一个相对容易验证的结果  
    assert fibonacci(10) == 55  
  
def test_fibonacci_negative_number():  
    # 假设我们想要函数对于负数的输入抛出异常或者返回特定的值  
    # 这里我们假设它应该返回0(或者你可以根据需求修改)  
    assert fibonacci(-1) == 0  

在项目目录运行 pytest命令,运行成功的结果如下图:



相关推荐
Swift社区1 小时前
在 Swift 中实现字符串分割问题:以字典中的单词构造句子
开发语言·ios·swift
没头脑的ht1 小时前
Swift内存访问冲突
开发语言·ios·swift
没头脑的ht1 小时前
Swift闭包的本质
开发语言·ios·swift
wjs20241 小时前
Swift 数组
开发语言
stm 学习ing2 小时前
FPGA 第十讲 避免latch的产生
c语言·开发语言·单片机·嵌入式硬件·fpga开发·fpga
湫ccc3 小时前
《Python基础》之字符串格式化输出
开发语言·python
mqiqe3 小时前
Python MySQL通过Binlog 获取变更记录 恢复数据
开发语言·python·mysql
AttackingLin3 小时前
2024强网杯--babyheap house of apple2解法
linux·开发语言·python