详解Python的单元测试


Posted in Python onApril 28, 2015

如果你听说过“测试驱动开发”(TDD:Test-Driven Development),单元测试就不陌生。

单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。

比如对函数abs(),我们可以编写出以下几个测试用例:

  •     输入正数,比如1、1.2、0.99,期待返回值与输入相同;
  •     输入负数,比如-1、-1.2、-0.99,期待返回值与输入相反;
  •     输入0,期待返回0;
  •     输入非数值类型,比如None、[]、{},期待抛出TypeError。

把上面的测试用例放到一个测试模块里,就是一个完整的单元测试。

如果单元测试通过,说明我们测试的这个函数能够正常工作。如果单元测试不通过,要么函数有bug,要么测试条件输入不正确,总之,需要修复使单元测试能够通过。

单元测试通过后有什么意义呢?如果我们对abs()函数代码做了修改,只需要再跑一遍单元测试,如果通过,说明我们的修改不会对abs()函数原有的行为造成影响,如果测试不通过,说明我们的修改与原有行为不一致,要么修改代码,要么修改测试。

这种以测试为驱动的开发模式最大的好处就是确保一个程序模块的行为符合我们设计的测试用例。在将来修改的时候,可以极大程度地保证该模块行为仍然是正确的。

我们来编写一个Dict类,这个类的行为和dict一致,但是可以通过属性来访问,用起来就像下面这样:

>>> d = Dict(a=1, b=2)
>>> d['a']
1
>>> d.a
1

mydict.py代码如下:

class Dict(dict):

  def __init__(self, **kw):
    super(Dict, self).__init__(**kw)

  def __getattr__(self, key):
    try:
      return self[key]
    except KeyError:
      raise AttributeError(r"'Dict' object has no attribute '%s'" % key)

  def __setattr__(self, key, value):
    self[key] = value

为了编写单元测试,我们需要引入Python自带的unittest模块,编写mydict_test.py如下:

import unittest

from mydict import Dict

class TestDict(unittest.TestCase):

  def test_init(self):
    d = Dict(a=1, b='test')
    self.assertEquals(d.a, 1)
    self.assertEquals(d.b, 'test')
    self.assertTrue(isinstance(d, dict))

  def test_key(self):
    d = Dict()
    d['key'] = 'value'
    self.assertEquals(d.key, 'value')

  def test_attr(self):
    d = Dict()
    d.key = 'value'
    self.assertTrue('key' in d)
    self.assertEquals(d['key'], 'value')

  def test_keyerror(self):
    d = Dict()
    with self.assertRaises(KeyError):
      value = d['empty']

  def test_attrerror(self):
    d = Dict()
    with self.assertRaises(AttributeError):
      value = d.empty

编写单元测试时,我们需要编写一个测试类,从unittest.TestCase继承。

以test开头的方法就是测试方法,不以test开头的方法不被认为是测试方法,测试的时候不会被执行。

对每一类测试都需要编写一个test_xxx()方法。由于unittest.TestCase提供了很多内置的条件判断,我们只需要调用这些方法就可以断言输出是否是我们所期望的。最常用的断言就是assertEquals():

self.assertEquals(abs(-1), 1) # 断言函数返回的结果与1相等

另一种重要的断言就是期待抛出指定类型的Error,比如通过d['empty']访问不存在的key时,断言会抛出KeyError:

with self.assertRaises(KeyError):
  value = d['empty']

而通过d.empty访问不存在的key时,我们期待抛出AttributeError:

with self.assertRaises(AttributeError):
  value = d.empty

运行单元测试

一旦编写好单元测试,我们就可以运行单元测试。最简单的运行方式是在mydict_test.py的最后加上两行代码:

if __name__ == '__main__':
  unittest.main()

这样就可以把mydict_test.py当做正常的python脚本运行:

$ python mydict_test.py

另一种更常见的方法是在命令行通过参数-m unittest直接运行单元测试:

$ python -m unittest mydict_test
.....
----------------------------------------------------------------------
Ran 5 tests in 0.000s

OK

这是推荐的做法,因为这样可以一次批量运行很多单元测试,并且,有很多工具可以自动来运行这些单元测试。
setUp与tearDown

可以在单元测试中编写两个特殊的setUp()和tearDown()方法。这两个方法会分别在每调用一个测试方法的前后分别被执行。

setUp()和tearDown()方法有什么用呢?设想你的测试需要启动一个数据库,这时,就可以在setUp()方法中连接数据库,在tearDown()方法中关闭数据库,这样,不必在每个测试方法中重复相同的代码:

class TestDict(unittest.TestCase):

  def setUp(self):
    print 'setUp...'

  def tearDown(self):
    print 'tearDown...'

可以再次运行测试看看每个测试方法调用前后是否会打印出setUp...和tearDown...。
小结

单元测试可以有效地测试某个程序模块的行为,是未来重构代码的信心保证。

单元测试的测试用例要覆盖常用的输入组合、边界条件和异常。

单元测试代码要非常简单,如果测试代码太复杂,那么测试代码本身就可能有bug。

单元测试通过了并不意味着程序就没有bug了,但是不通过程序肯定有bug。

Python 相关文章推荐
python smtplib模块发送SSL/TLS安全邮件实例
Apr 08 Python
python解析xml文件实例分析
May 27 Python
基于进程内通讯的python聊天室实现方法
Jun 28 Python
Python进阶篇之字典操作总结
Nov 16 Python
使用Python写一个小游戏
Apr 02 Python
python使用magic模块进行文件类型识别方法
Dec 08 Python
基于Python的图像数据增强Data Augmentation解析
Aug 13 Python
Python中的引用和拷贝实例解析
Nov 14 Python
python二分法查找算法实现方法【递归与非递归】
Dec 06 Python
浅析python 定时拆分备份 nginx 日志的方法
Apr 27 Python
Python爬虫教程之利用正则表达式匹配网页内容
Dec 08 Python
Python利用imshow制作自定义渐变填充柱状图(colorbar)
Dec 10 Python
Python xlrd读取excel日期类型的2种方法
Apr 28 #Python
Python发送email的3种方法
Apr 28 #Python
Python中使用partial改变方法默认参数实例
Apr 28 #Python
调试Python程序代码的几种方法总结
Apr 28 #Python
解析Python中的异常处理
Apr 28 #Python
python调用java模块SmartXLS和jpype修改excel文件的方法
Apr 28 #Python
Python EOL while scanning string literal问题解决方法
Sep 18 #Python
You might like
PHP5+UTF8多文件上传类
2008/10/17 PHP
Windows下的PHP 5.3.x安装 Zend Guard Loader教程
2014/09/06 PHP
php中print(),print_r(),echo()的区别详解
2014/12/01 PHP
PHP中使用imagick生成PSD文件缩略图教程
2015/01/26 PHP
浅谈PHP中静态方法和非静态方法的相互调用
2016/10/04 PHP
jQuery+php简单实现全选删除的方法
2016/11/28 PHP
PHP简单实现二维数组赋值与遍历功能示例
2017/10/19 PHP
php实现生成带二维码图片并强制下载功能
2018/02/24 PHP
禁止JQuery中的load方法装载IE缓存中文件的方法
2009/09/11 Javascript
JavaScript异步调用定时方法并停止该方法实现代码
2012/03/16 Javascript
js父窗口关闭时子窗口随之关闭完美解决方案
2014/04/29 Javascript
IE6中链接A的href为javascript协议时不在当前页面跳转
2014/06/05 Javascript
node.js中的fs.fchmod方法使用说明
2014/12/16 Javascript
js中的内部属性与delete操作符介绍
2015/08/10 Javascript
基于bootstrap插件实现autocomplete自动完成表单
2016/05/07 Javascript
AngularJS中的指令全面解析(必看)
2016/05/20 Javascript
Node之简单的前后端交互(实例讲解)
2017/11/14 Javascript
微信小程序引入模块中wxml、wxss、js的方法示例
2019/08/09 Javascript
[01:36]极致酷炫!TI9典藏宝瓶+撼地者至宝展示
2019/06/11 DOTA
Python的collections模块中namedtuple结构使用示例
2016/07/07 Python
简单了解python协程的相关知识
2019/08/31 Python
Windows 平台做 Python 开发的最佳组合(推荐)
2020/07/27 Python
蔻驰意大利官网:COACH意大利
2019/01/16 全球购物
马德里运动鞋商店:Nigra Mercato
2020/02/16 全球购物
泰国最新活动和优惠:Megatix
2020/05/07 全球购物
护理自我鉴定范文
2013/10/06 职场文书
小学生自我鉴定
2013/10/12 职场文书
办公室年终个人自我评价
2013/10/28 职场文书
计算机应用与科学个人的自我评价
2013/11/15 职场文书
法学个人求职信范文
2014/01/27 职场文书
北京申奥口号
2014/06/19 职场文书
大学生个人求职信例文
2014/07/07 职场文书
学生吸烟检讨书
2014/09/14 职场文书
万能检讨书2000字
2014/10/17 职场文书
机关作风建设自查报告
2014/10/22 职场文书
Java 多线程并发FutureTask
2022/06/28 Java/Android