详解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 open读写文件实现脚本
Sep 06 Python
Python中转换角度为弧度的radians()方法
May 18 Python
Python中用字符串调用函数或方法示例代码
Aug 04 Python
Python实现学校管理系统
Jan 11 Python
django 在原有表格添加或删除字段的实例
May 27 Python
对Python中列表和数组的赋值,浅拷贝和深拷贝的实例讲解
Jun 28 Python
使用11行Python代码盗取了室友的U盘内容
Oct 23 Python
python实现两张图片拼接为一张图片并保存
Jul 16 Python
Python 使用多属性来进行排序
Sep 01 Python
python多环境切换及pyenv使用过程详解
Sep 27 Python
Python猫眼电影最近上映的电影票房信息
Sep 18 Python
Python在centos7.6上安装python3.9的详细教程(默认python版本为2.7.5)
Oct 15 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
php array_intersect比array_diff快(附详细的使用说明)
2011/07/03 PHP
php+ajax实现的点击浏览量加1
2015/04/16 PHP
PHP实现XML与数据格式进行转换类实例
2015/07/29 PHP
PHP中spl_autoload_register()函数用法实例详解
2016/07/18 PHP
PHP随机获取未被微信屏蔽的域名(微信域名检测)
2017/03/19 PHP
TNC vs IO BO3 第二场2.13
2021/03/10 DOTA
15个款优秀的 jQuery 图片特效插件推荐
2011/11/21 Javascript
javascript使用window.open提示“已经计划系统关机”的原因
2014/08/15 Javascript
node.js中的fs.truncateSync方法使用说明
2014/12/15 Javascript
js文字横向滚动特效
2015/11/11 Javascript
Webpack 实现 AngularJS 的延迟加载
2016/03/02 Javascript
JS控制div跳转到指定的位置的几种解决方案总结
2016/11/05 Javascript
JS实现含有中文字符串的友好截取功能分析
2017/03/13 Javascript
webpack 1.x升级过程中的踩坑总结大全
2017/08/09 Javascript
浅谈Vue Element中Select下拉框选取值的问题
2018/03/01 Javascript
Vue + better-scroll 实现移动端字母索引导航功能
2018/05/07 Javascript
简单的三步vuex入门
2018/05/20 Javascript
微信小程序swiper实现文字纵向轮播提示效果
2020/01/21 Javascript
JS出现404错误原理及解决方案
2020/07/01 Javascript
Vue3.0的优化总结
2020/10/16 Javascript
[02:50]2014DOTA2 TI预选赛预选赛 大神专访第一弹!
2014/05/21 DOTA
[36:09]Secret vs VG 2019国际邀请赛淘汰赛 败者组 BO3 第一场 8.24
2019/09/10 DOTA
Python tempfile模块学习笔记(临时文件)
2014/05/25 Python
Python代码调试的几种方法总结
2015/04/15 Python
python探索之BaseHTTPServer-实现Web服务器介绍
2017/10/28 Python
python 字典有序并写入json文件过程解析
2019/09/30 Python
Python sqlite3查询操作过程解析
2020/02/20 Python
浅谈Python 参数与变量
2020/06/20 Python
Django基于Models定制Admin后台实现过程解析
2020/11/11 Python
CSS3实战第一波 让我们尽情的圆角吧
2010/08/27 HTML / CSS
CSS3 实现侧边栏展开收起动画
2014/12/22 HTML / CSS
美国演唱会和体育门票购买网站:Ticketnetwork
2018/10/19 全球购物
系统管理员的职责包括那些?管理的对象是什么?
2016/09/20 面试题
创先争优活动方案
2014/02/12 职场文书
python周期任务调度工具Schedule使用详解
2021/11/23 Python
python如何为list实现find方法
2022/05/30 Python