单元测试

如果你听过“测试驱动开发”（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。