详解Python之unittest单元测试代码

2020-01-04 16:05:46

字体：大中小

来源：转载

供稿：网友

前言

编写函数或者类时，还可以为其编写测试。通过测试，可确定代码面对各种输入都能够按要求的那样工作。

本次我将介绍如何使用python/293344.html">Python模块unittest中的工具来测试代码。

测试函数

首先我们先编写一个简单的函数，它接受姓、名、和中间名三个参数，并返回完整的姓名：

names.py

def get_fullname(firstname,lastname,middel=''):  '''创建全名'''  if middel:    full_name = firstname + ' ' + middel + ' ' + lastname    return full_name.title()  else:    full_name = firstname + ' ' + lastname    return full_name.title()

然后再当前目录下编写调用函数程序

get_name.py

from names import get_fullnamemessage = "Please input 'q' to quit."print(message)while True:  first = input("Please input your firstname: ")  if first == 'q':    break  last = input("Please input your lastname: ")  if last == 'q':    break  middels = input("Please input your middel name or None: ")  if last == 'q':    break  formant_name = get_fullname(first,last,middels)  print("/tYour are fullname is: " + formant_name.title())

调用结果：

Please input 'q' to quit.
Please input your firstname: xiao
Please input your lastname: peng
Please input your middel or None:
Your are fullname is: Xiao Peng
Please input your firstname: xiao
Please input your lastname: peng
Please input your middel or None: you
Your are fullname is: Xiao You Peng
Please input your firstname: q
进程已结束,退出代码0

创建测试程序

创建测试用例的语法需要一段时间才能习惯，但测试用例创建后，再针对函数的单元测试就很简单了。先导入模块unittest以及要测试的函数，再创建一个继承函数unittest.TestCase的类，

并编写一系列方法对函数行为的不同方便进行测试。

下面介绍测试上面names.py函数是否能够正确的获取姓名：

Test_get_name.py

import unittestfrom names import get_fullnameclass NamesTestCase(unittest.TestCase):  '''定义测试类'''  def test_get_name2(self):    '''测试2个字的名字'''    formatied_name2 = get_fullname('xiao','pengyou')    self.assertEqual(formatied_name2,'Xiao Pengyou')  def test_get_name3(self):    '''测试3个字的名字'''    formatied_name3 = get_fullname('xiao','peng',middel='you')    self.assertEqual(formatied_name3,'Xiao Peng You')if __name__ == '__init__':  unittest.main()

测试结果：

Ran 2 tests in 0.034s

OK

两个测试单元测试通过测试！

在当前的大目录下会生成一个测试报告，可以通过浏览器进行打开查看。

Python,unittest单元测试,unittest

由图可知，两个测试通过，并显示测试的时间！！！

unittest.TestCase的各种断言方法

unittest各种断言方法

方法	用途
assertEqual(a,b)	核实a == b
assertNotEqual(a,b)	核实a != b
assertTrue(x)	核实x为True
assertFalse(x)	核实x为False
assertIn(item,list)	核实item在list中
assertNotIn(item,list)	核实item不在list中