Python 测试框架之 Unittest & Pytest

2022-07-04 17:07:48 浏览数 (1)

Unittest

Unittest是Python自带的一个单元测试框架

Unittest中包含了对一些常规的测试用例体系性的支持,主要包括:

  • test fixture
  • test case
  • test suite
  • test runner

创建.py文件,并选择文件类型如下图所示:

点击OK后默认生成的文件内容如下:

可以在此文件的内容基础上进行改造,写自己的测试case,改造后的内容如下:

代码语言:javascript复制
import unittest  # 导入unittest模块

def number_sum(a, b):
    return a   b

"""
setUp():每个测试case运行之前运行
tearDown():每个测试case运行完之后执行
setUpClass():必须使用@classmethod 装饰器,  所有case运行之前只运行一次
tearDownClass():必须使用@classmethod装饰器, 所有case运行完之后只运行一次

"""

class MyTestCase(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("setupClass")

    def setUp(self):
        print("setUp")

    def test_sum_int(self):
        self.assertEqual(number_sum(1, 2), 3)
        self.assertEqual(number_sum(100, 300), 400)

    def test_sum_number(self):
        self.assertEqual(number_sum(1.1, 2.2), 3.3)

    def tearDown(self):
        print("tearDown")

    @classmethod
    def tearDownClass(cls):
        print("tearDownClass")


if __name__ == '__main__':
    unittest.main()

以上代码是标准的unittest测试框架中的基础结构。

Pytest

Pytest中定义测试用例主要有三种方式:

  • 兼容Unittest,已有的Unittest库和文件都可以直接进行调用
  • 可以基于最简单的类进行定义,如果一个类里面没有初始化方法并且以Test开头,系统便会认为这是一个测试用例
  • 可以直接定义测试函数来定义测试用例

安装Pytest: 1、 使用命令:pip install pytest 2、 在Pycharm编译器里的配置中添加,如下截图所示

Pytest安装成功后,将之前默认的runner由unittest修改为:pytest,修改步骤如下截图所示:

将Pytest配置成默认的runner后,新建一个.py文件为:test_demos.py 内容如下:

点击如上图所示的绿色小箭头,可查看到第一行提示:使用Pytest来运行test_demos.py文件,运行结果正常

将Python集成工具由Unittest切换成Pytest后,再来运行之前写的使用Unittest框架编写的代码,如图所示:

类名前的绿色小箭头提示可以使用:Pytest来运行test_unittest.py文件,Pytest兼容Unittest,可以直接使用Pytest直接调用Unittest库和文件。运行结果如下所示:

Pytest的用例识别规则:

  • Test类包含的所有test_方法
  • 不在class中的所有test_*函数
  • 类中不能初始化方法
  • 断言使用基本的assert即可

文件范围:

  • test_*.py
  • *_test.py

用例执行顺序的控制方法如下图所示:

完整代码如下:

代码语言:javascript复制
import pytest #导入pytest模块

def func(x):
    return x   1

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


class TestFunc:
    @classmethod
    def setup_class(self):
        print("setup_class")

    def setup(self):
        print("setup")

    @pytest.mark.fail
    def test_answer1(self):
        print("test_answer1")
        assert func(3) == 5

    @pytest.mark.success
    @pytest.mark.parametrize("input,expect",{
        (5, 6),
        (7, 8),
        (0, 1),
        (2, 2)
    })
    def test_answer2(self, input, expect):
        print("test_answer2")
        assert func(input) == expect

    def test_answer3(self):
        print("test_answer3")
        assert func(7) == 8

    def teardown(self):
        print("teardown")

    @classmethod
    def teardown_class(cls):
        print("teardown_class")

Pytest支持分组: 1、@pytest.mark.webtest 2、@pytest.mark.sec 3、Pytest -m “webtest and not sec”

调用方法如下图所示:

在Terminal里输入命令:pytest -m fail来实现只执行指定的用例case的目的。 参数化用例的使用方法的实例代码如下所示:

代码语言:javascript复制
@pytest.mark.parametrize("input,expect",{
    (5, 6),
    (7, 8),
    (0, 1),
    (2, 2)
})
def test_answer2(self, input, expect):
    print("test_answer2")
    assert func(input) == expect

参数化用例方法可以达到代码重复利用的效果。 单独来执行test_answer2这个用例方法时会发现实际共执行了4次该方法,input与expect的值分别是参数化中的(5, 6),(7, 8),(0, 1),(2, 2),结果如下:

总结:今天分享的内容是Python测试框架中常用的框架Unittest与Pytest相关内容,在自动化测试过程中可以进行实操,提高测试效率。 下一期将分享使用allure来实现测试报告的可视化展示,敬请期待~

友情提示:“无量测试之道”原创著作,欢迎关注交流,禁止第三方不显示文章来源时转载。

0 人点赞