15. Unittest单元测试框架的介绍与使用

Posted suim1218

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了15. Unittest单元测试框架的介绍与使用相关的知识,希望对你有一定的参考价值。

?什么是单元测试

在计算机编程中,单元测试(英语:Unit Testing)又称为模块测试, 是针对程序模块(软件设计的最小单位)来进行正确性检验的测试工作。程序单元是应用的最小可测试部件。在过程化编程中,一个单元就是单个程序、函数、过程等;对于面向对象编程,最小单元就是方法,包括基类(超类)、抽象类、或者派生类(子类)中的方法。

上面这段内容摘自维基百科。笔者在入职的时候,一直搞不明白什么是单元测试。即使搜了很多资料,也还是搞不明白。一直到接触代码的时候,才对单元测试有了一个大概的认识。当然以前也一直有一个误解,觉得单元测试是开发做的,跟测试同学扯不上关系,直到真正从事自动化工作时,才知道原来单元测试不管是开发还是测试同学都可以做。

可能到这里一些读者还是不明白什么是单元测试,笔者将用自己的理解来解释一下什么是单元测试。同时下面也会给出例子让大家更深入了解什么是单元测试。之前我们讲Python基础的时候,讲到如何定义一个函数。比如我们现在有一个加法函数,需要我们输入两个值,那么我们怎么保证我们写的函数是没有问题的呢?是不是要调用这个函数,然后传入两个值,查看返回值是否正确?其实这就是在做单元测试。当然我们没有用规范的做法去做单元测试,那么什么是规范的方法呢?下面我将会以代码形式举例进行介绍。还是用加法函数来进行举例。如下面代码所示:

def add(a, b):
????return a + b

print(add(1, 3))

我们已经定义一个add函数,现在调用add函数,传入1和3,然后把返回值进行打印,通过打印我们发现打印的结果为4,可以初步这个函数是没问题的,但是我们知道一组数据不能证明这个函数是没错的,所以我们需要多验证几次,这个时候我们又加入了一行测试代码,如下面代码所示:

def add(a, b):

????return a + b

print(add(1, 3))
print(add('a', 3))

运行之后我们发现我们的程序报错了,报错内容如下:

Traceback (most recent call last):
??File "F:/blogApiTest/test2.py", line 6, in
????print(add(‘a‘, 3))
??File "F:/blogApiTest/test2.py", line 2, in add
????return a + b
TypeError: must be str, not int

原来字符串和整型不能相加!当然我们大家都知道在Python程序里字符串和整型不能相加,但是我们就是要传入字符串和整型进行相加,结果程序奔溃了。所以我们可以说这个add函数是有bug的,那么我们来修复一下上面代码,如下面代码所示:

def add(a, b):
????if (type(a) == int or type(a) == float) and (type(b) == int or type(b) == float):
????????return a + b
????else:
????????return "请输入数字"

print(add(1, 3))
print(add('a', 3))

通过测试后我们发现程序已经被我们修复,但是这里我们思考一下,我们在编写测试用例的时候都要写一个预期结果,但是代码中并没有写预期结果,预期结果在我们的脑海里,我们根据打印出来的实际结果跟我们脑海中预期结果进行比对,如果一致说明程序是没问题的,如果不一致,程序是有问题的。想想看,其实挺麻烦的,那么我们有没有办法在代码中写上预期结果和实际结果呢?答案是肯定的,有!使用assert断言!

加入assert断言

assert一般格式是assert condition?, 如果condition为false,那么raise一个AssertionError出来。
比如我们断言1和2是否相等,可以写成 assert 1==2 。知道这个逻辑我们就可以给我们的add函数加上断言,如下代码所示:

def add(a, b):
????if type(a) == int and type(b) == int:
????????return a + b
????else:
????????return "请输入数字"

assert add(1, 2) == 3
assert add(1, 3) == 3

运行之后我们会发现程序报出异常,如下代码所示:

Traceback (most recent call last):
??File "F:/blogApiTest/test2.py", line 9, in
????assert add(1, 3) == 3
AssertionError

通过上面异常我们发现,如果预期结果和实际结果不一致时,assert断言只抛出 AssertionError,这样我们很难搞清断言为什么失败,对于我们排查错误时也造成不必要的时间浪费。那么有没有别的方法可以很快知道为什么会断言失败呢?答案是有的,使用Python Unittest单元的测试框架。

unittest单元测试框架

先贴一下unittest官网地址。unittest文档开头介绍了四个重要的概念:test fixture,test case,?test suite,?test?runner。

test fixture

test fixture表示执行一个或多个测试所需的准备。怎么去理解这句话呢?比如我们执行一条创建文章的测试用例,但是如果我们调用创建文章的接口,需要我们进行登录。那么我们可以把登录作为一个test fixture。那么test fixture表现形式是什么呢?还是用add函数举例。先说test fixture中有几个

test case

测试用例,这个不必多说,没有测试用例就不能叫做接口测试了。

test suite

我们在做接口测试的时候,一般都是执行多条测试用例。这就有了test suite概念,简单来说,就是把想要执行的测试用例进行组装然后进行测试。

test ?runner

执行测试用例。
为了方便大家的理解,会用一个实例对上面四个概念进行解释。请看如下代码:

import unittest


class AddTest(unittest.TestCase):
    def setUp(self):
        print("在每个用例执行之前")

    def tearDown(self):
        print("在每个用例执行之后")

    @classmethod
    def setUpClass(cls):
        print("在用例执行之前,只执行一次,优先于setUp")

    @classmethod
    def tearDownClass(cls):
        print("所有用例运行之后只执行一次,优先于tearDown")

    def add(self, a, b):
        if type(a) == int and type(b) == int:
            return a + b
        else:
            return "请输入数字"

    def test_add1(self):
        self.assertEqual(self.add(2, 3), 5, "验证加法")

    def test_add2(self):
        self.assertEqual(self.add(2, 4), 6, "验证加法")


if __name__ == '__main__':
    # unittest.main()
    # test suite(构造测试集)
    suite = unittest.TestSuite()
    suite.addTest(AddTest("test_add1"))
    suite.addTest(AddTest("test_add2"))
    # test runner?(执行测试)
    runner = unittest.TextTestRunner()
    runner.run(suite)

上面代码输出如下内容:

在用例执行之前,只执行一次,优先于setUp所有用例运行之后只执行一次,优先于tearDown在每个用例执行之前
在每个用例执行之后
在每个用例执行之前
在每个用例执行之后

我们来根据代码拆解一下,看哪部分对应test fixture,test case,?test suite,?test?runner。首先,在使用unittest单元测试框架之前,我们需要导包,我们先import unittest,然后创建一个AddTest类,这个类继承unittest.TestCase。然后我们创建一个add方法(注意:一般类中的函数我们称作为方法),然后创建了2条测试用例test_add1,test_add2,注意!测试用例必须以test开头!然后使用test fixture中setUp、tearDown、setUpClass 、tearDownClass。然后使用test suite把2条测试用例加进去,最后使用test runner执行测试用例。到这里想必大家都已经明白。最后画一张思维导图,更方便大家的理解。

技术图片

细心的人可以看到,测试用例中有一个assertEqual,这就是我们说比assert更好用的断言,那么这些断言有哪些呢?请看下面表格:

Method Checks that
assertEqual(a,?b) a?==?b
assertNotEqual(a,?b) a?!=?b
assertTrue(x) bool(x)?is?True
assertFalse(x) bool(x)?is?False
assertIs(a,?b) a?is?b
assertIsNot(a,?b) a?is?not?b
assertIsNone(x) x?is?None
assertIsNotNone(x) x?is?not?None
assertIn(a,?b) a?in?b
assertNotIn(a,?b) a?not?in?b
assertIsInstance(a,?b) isinstance(a,?b)
assertNotIsInstance(a,?b) not?isinstance(a,?b)

小结:本章内容只是简单介绍了一下unittest的使用方法以及相关概念,unittest框架使用起来很简单也很方便,建议大家多去官网学习,当然不管是UI自动化也好,还是接口自动化也好,如果使用Python实现,一般都不会绕过Unittest这个概念,希望大家可以熟练掌握。

以上是关于15. Unittest单元测试框架的介绍与使用的主要内容,如果未能解决你的问题,请参考以下文章

Python接口自动化测试之pytest与unittest区别

unittest介绍

Python单元测试框架unittest之深入学习

单元测试框架之unittest

Python单元测试:unittest使用简介

单元测试框架之unittest