python 单元测试 工具_Python单元测试工具doctest和unittest

Python标准库包含两个测试工具。

doctest：一个简单的模块，为检查文档而设计，但也适合用来编写单元测试。

unittest：一个通用的测试框架。

一、使用doctest进行单元测试

创建文件mymath.py，内容

defsquare(x):'''计算平方并返回结果(下面是单元测试的格式)

>>> square(2)

4

>>> square(3)

9'''

return x *xif __name__ == '__main__':importdoctest,mymath

doctest.testmod(mymath)

在Sublime Text 3中运行只是提示[Finished in 0.2s]。

在cmd命令下切换到mymath.py所在目录，运行python mymath.py -v后，有提示测试详细信息，测试通过。如下图：

把函数square里面的return x * x 改成 return x / x。

再次运行python mymath.py -v，提示测试不通过，如下图：

二、使用unittest进行单元测试

还是用mymath.py测试，内容：

defsquare(x):return x * x

新建单元测试文件test_math.py，内容：

importunittest, mymathclassmathTestCase(unittest.TestCase):deftest_square(self):

self.assertEqual(mymath.square(2), 4)

self.assertEqual(mymath.square(3), 9)if __name__ == '__main__':

unittest.main()

运行后，显示OK 测试通过。

.

----------------------------------------------------------------------

Ran 1 test in 0.000s

OK

把函数square里面的return x * x 改成 return x / x。

再次运行，这次显示FAILED 测试不通过。

F

======================================================================

FAIL: test_square (__main__.mathTestCase)

----------------------------------------------------------------------

Traceback (most recent call last):

File "D:\projects\python\test_math.py", line 6, in test_square

self.assertEqual(mymath.square(2), 4)

AssertionError: 1.0 != 4

----------------------------------------------------------------------

Ran 1 test in 0.000s

FAILED (failures=1)