代码写多了越发觉得测试的重要性，之前一直喜欢“目测”的做法已经不值得推荐了。当然，这只是一个玩笑。
在Python代码里测试大概有这么几种：doctest、unittest和nose（第三方工具）。
个人推荐nose，简单的话doctest也已经足够了。
首先：
- 代码要易于测试，代码写完对应的测试应该配套跟上。
- 测试要简单，轻便。
先说最简单的doctest吧，顾名思义，doctest就是在文档（docstring）里完成测试。看以下例子（/tmp/1.py）：
def add(a, b):
"""
>>> add(10, 20)
30
"""
return a + b
if __name__ == '__main__':
import doctest
doctest.testmod()
可以看出doctest是不是很简单？是不是想起了普通解释器（>>>），运行一下测试（注意-v选项）：
smallfish@debian:~$ python /tmp/1.py -v
Trying:
add(10, 20)
Expecting:
30
ok
1 items had no tests:
__main__
1 items passed all tests:
1 tests in __main__.add
1 tests in 2 items.
1 passed and 0 failed.
Test passed.
作为简单的模块或者工具，main里实现测试已经足够了。但是稍大一些工程或应用，doctest太多的话，看上去会觉得有点臃肿。
下面介绍下nose，偷偷的说下，nose是支持doctest、unittest测试的，所以嘛。先安装一下：
smallfish@debian:~$ pip install nose
继续跑一下上面的例子：
smallfish@debian:~$ nosetests /tmp/1.py --with-doctest -v Doctest: 1.add ... ok ---------------------------------------------------------------------- Ran 1 test in 0.006s OK
好吧，写点常见的例子，目录结构如下（foo模块以及foo的测试代码）：
smallfish@debian:~$ tree /tmp/foomodule/
/tmp/foomodule/
|-- foo
| |-- a.py
| |-- b.py
| `-- __init__.py
`-- tests
|-- test_a.py
`-- test_b.py
模块代码如下：
# /tmp/foomodule/foo/a.py
def add(a, b):
return a + b
def double(a):
return a * 2
# /tmp/foomodule/foo/b.py
import memcache
class Cache:
def __init__(self, server):
self.cache = memcache.Client([server])
def get(self, name):
return self.cache.get(name)
def set(self, name, value):
return self.cache.set(name, value)
def delete(self, name):
return self.cache.delete(name)
def close(self):
self.cache.disconnect_all()
对应的测试代码：
# /tmp/foomodule/tests/test_a.py
from foo.a import add, double
def test_add():
v = add(10, 20)
assert v == 30
def test_double():
v = double(10)
assert v == 20
# /tmp/foomodule/tests/test_b.py
from foo.b import Cache
class TestCache:
def setUp(self):
self.cache = Cache("127.0.0.1:11211")
self.key = "name"
self.value = "smallfish"
def tearDown(self):
self.cache.close()
def test_00_get(self):
v = self.cache.get(self.key)
assert v == None
def test_01_set(self):
v = self.cache.set(self.key, self.value)
assert v == True
v = self.cache.get(self.key)
assert v == self.value
def test_02_delete(self):
v = self.cache.delete(self.key)
assert v == True
先不解释，跑一跑（-v -s选项，针对当前路径下tests目录测试）：
smallfish@debian:/tmp/foomodule$ nosetests -s -v test_a.test_add ... ok test_a.test_double ... ok test_b.TestCache.test_00_get ... ok test_b.TestCache.test_01_set ... ok test_b.TestCache.test_02_delete ... ok ---------------------------------------------------------------------- Ran 5 tests in 0.024s OK
简单吧？nose默认是当前目录下tests目录进行测试，默认规则：文件（含有test），函数（test_开头），类（Test开头）。
nose完整文档请参考：http://nose.readthedocs.org/en/latest/
转载请注明：爱开源 » Python Testing