python testing_Python Testing

代码写多了越发觉得测试的重要性，之前一直喜欢“目测”的做法已经不值得推荐了。当然，这只是一个玩笑。

在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 testin 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开头)。

END