Unittest

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

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

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

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

Python系统测试框架_Python系统测试框架

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

Python系统测试框架_软件测试_02

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

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编译器里的配置中添加，如下截图所示

Python系统测试框架_python_03

Python系统测试框架_软件测试_04

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

Python系统测试框架_pytest_05

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

Python系统测试框架_Python系统测试框架_06

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

Python系统测试框架_Python系统测试框架_07

Python系统测试框架_自动化测试_08

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

Python系统测试框架_python_09

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

Python系统测试框架_pytest_10

Pytest的用例识别规则：

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

文件范围：

  • test_*.py 
  • *_test.py

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

Python系统测试框架_python_11

完整代码如下：

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”

调用方法如下图所示：

Python系统测试框架_Python系统测试框架_12

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

@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系统测试框架_Python系统测试框架_13

总结：今天分享的内容是Python测试框架中常用的框架Unittest与Pytest相关内容，在自动化测试过程中可以进行实操，提高测试效率。