Python和单元测试那些事儿(2)

当前位置:

首页 > Python基础教程 >

Python和单元测试那些事儿(2)

测试结果：

1

2

3

root@arch tests: python test_demo.py

DEBUG:root:start of test...

DEBUG:root:end of test...

unittest

这个文档确实有点长，我感觉还是仔细去读一下文档比较好（虽然我也没读完）。

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):

        self.alist = []

    def tearDown(self):

        print(self.alist)

    def test_list(self):

        for i in range(5):

            self.alist.append(i)

if __name__ == '__main__':

    unittest.main()

1

2

3

4

5

root@arch tests: python test_demo.py

[0, 1, 2, 3, 4]

.

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

Ran 1 test in 0.001s

unittest框架配合上Mock，单元测试基本无忧啦。

pytest

上面的单元测试跑起来比较麻烦，当然也可以写一个脚本遍历所有的单元测试文件，然后执行。不过 pytest 对unittest有比较好的支持。

pytest默认支持的是函数风格的测试，但是我们可以不用这一块嘛（而且很多时候还是很有用的）。走进项目根目录，输入 pytest 就可以啦。它会自动发现 test_ 开头的文件，然后执行其中 test_ 开头的函数和 unittest 的 test_ 开头的方法。

root@arch tests: pytest

========================================================= test session starts =========================================================

platform linux -- Python 3.5.2, pytest-3.0.5, py-1.4.31, pluggy-0.4.0

rootdir: /root/tests, inifile:

collected 1 items

test_afunc.py .

====================================================== 1 passed in 0.03 seconds =======================================================

root@arch tests:

栏目列表