python测验6_Python 6.2 单元测试

单元测试

如果你听过“测试驱动开发”（TDD:Test_Driven Development），单元测试你就不会陌生。

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

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

1、输入正数

2、输入负数

3、输入0

4、输入非数值类型

把上面的测试用例放在一个模块中，就是一个完整的单元测试。

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

单元测试有什么意义？如果我们对abs()函数代码做了修改，只要再跑一边单元测试，如果通过，说明我们的修改不会对abs()函数原有的行为造成影响，如果测试不通过，说明我们的修改与原有的行为不一致，用起来就像下面这样:>>> d = Dict(a=1, b=2)

>>> d['a']

1

>>> d.a

1

mydict.py代码如下:

class Dict(dict):

def __init__(self,**kw):

super().__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 Test_Dict(unittest,TestCase):

def test_init(self):

d =Dict(a =1,b ='test')

self.assertEqual(d.a,1)

self.assertEqual(d.b,'test')

self.assertTrue(inistance(d,dict))

def test_key(self):

d =Dict()

d['key'] ='value'

self.assertEqual(d.key,'value')

def test_attr(self):

d =Dict()

d.key ='value'

self.assertTrue('key',in d)

self.assertEqual(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提供了很多内置的条件判断，我们只需要调用这些方法就可以断言输出是否使我们需要的。最常用的断言就是assertEqual():

self.assertEqual(abs(-1),1)。

另外一种断言是期待抛出制定类型的错误，比如通过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脚本运行:$ python3 mydict_test.py

另一种方法是在命令行通过参数 -m unittest直接运行测试单元测试:$ python3 -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。