前言
编写函数或者类时,还可以为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。本次我将介绍如何使用Python模块unittest中的工具来测试代码。
测试函数
首先我们先编写一个简单的函数,它接受姓、名、和中间名三个参数,并返回完整的姓名:
names.py
def get_fullname(firstname,lastname,middel=''):'''创建全名'''
ifmiddel:
full_name= firstname + ' ' + middel + ' ' +lastnamereturnfull_name.title()else:
full_name= firstname + ' ' +lastnamereturn full_name.title()
然后再当前目录下编写调用函数程序
get_name.py
from names importget_fullname
message= "Please input 'q' to quit."
print(message)whileTrue:
first= input("Please input your firstname:")if first == 'q':breaklast= input("Please input your lastname:")if last == 'q':breakmiddels= input("Please input your middel name or None:")if last == 'q':breakformant_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 middelorNone:
Your are fullnameis: Xiao Peng
Please input your firstname: xiao
Please input your lastname: peng
Please input your middelorNone: you
Your are fullnameis: Xiao You Peng
Please input your firstname: q
进程已结束,退出代码0
创建测试程序
创建测试用例的语法需要一段时间才能习惯,但测试用例创建后,再针对函数的单元测试就很简单了。先导入模块unittest以及要测试的函数,再创建一个继承函数unittest.TestCase的类,
并编写一系列方法对函数行为的不同方便进行测试。
下面介绍测试上面names.py函数是否能够正确的获取姓名:
Test_get_name.py
importunittestfrom names importget_fullnameclassNamesTestCase(unittest.TestCase):'''定义测试类'''
deftest_get_name2(self):'''测试2个字的名字'''formatied_name2= get_fullname('xiao','pengyou')
self.assertEqual(formatied_name2,'Xiao Pengyou')deftest_get_name3(self):'''测试3个字的名字'''formatied_name3= get_fullname('xiao','peng',middel='you')
self.assertEqual(formatied_name3,'XiaoPeng You')if __name__ == '__init__':
unittest.main()
测试结果:
Ran 2 tests in0.034s
OK
两个测试单元测试通过测试!
在当前的大目录下会生成一个测试报告,可以通过浏览器进行打开查看。
由图可知,两个测试通过,并显示测试的时间!!!
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中