前言
编写函数或者类时,还可以为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。
本次我将介绍如何使用Python模块unittest中的工具来测试代码。
测试函数
首先我们先编写一个简单的函数,它接受姓、名、和中间名三个参数,并返回完整的姓名:
names.py
1
2
3
4
5
6
7
8
|
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
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from names import get_fullname
message = "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.
进程已结束,退出代码0
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
创建测试程序
创建测试用例的语法需要一段时间才能习惯,但测试用例创建后,再针对函数的单元测试就很简单了。先导入模块unittest以及要测试的函数,再创建一个继承函数unittest.TestCase的类,
并编写一系列方法对函数行为的不同方便进行测试。
下面介绍测试上面names.py函数是否能够正确的获取姓名:
Test_get_name.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
import unittest
from names import get_fullname
class 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
两个测试单元测试通过测试!
在当前的大目录下会生成一个测试报告,可以通过浏览器进行打开查看。
由图可知,两个测试通过,并显示测试的时间!!!
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中 |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:http://www.cnblogs.com/JeremyWYL/p/8340316.html