如何通过unittest模块在Python中执行单元测试用例？

本文共计1350个文字，预计阅读时间需要6分钟。

Python unittest 框架为我们提供了多种执行测试用例的方法。在这里，我们来学习一些常用操作。

- 脚本自测：使用 unittest.main() 会自动收集当前文件中所有的测试用例并执行。

​

---

​​unittest​​框架给我们准备了多种执行测试用例的方法。这里我们来学习一些经常用到的操作。

脚本自测

​​unittest.main​​会自动收集当前文件中所有的测试用例来执行：

import unittest

blog_name = 'hy592070616'

class MyTestClass(unittest.TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')

def setUp(self):
self.blog_name = 'hy592070616'
print('setUp')

def test_first(self):
blog_name_ref = blog_name + '--machine learning'
self.assertEqual(blog_name_ref, 'hy592070616--machine learning')


def tearDown(self):
print('tearDown')

@classmethod
def tearDownClass(cls):
print('tearDownClass')


if __name__ == '__main__':
unittest.main()

• ​​if __name__ == '__main__':​​​的​​name​​​和​​main​​​前后都有两个下划线。我们可以在PyCharm中直接输入​​main​​来快速输入这行代码。
• ​​unittest.main()​​​，​​main​​后面需要加上括号，否则无法正常执行。
• ​​unittest.main()​​，是对上方脚本进行自测，不影响其他文件的调用。

通过class构造测试集合

我们可以通过​​unittest.TestLoader().loadTestsFromTestCase​​​加载某个​​class​​​下面的所有用例。这个是我们的一个测试用例文件​​testclass.py​​​，包含了4个测试用例和2个​​class​​：

# testclass.py
import unittest


class MyTestClassOne(unittest.TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')

def setUp(self):
print('setUp')

def test_first(self):
print('test_first')
self.assertEqual(0, 0)

def test_second(self):
print('test_second')
self.assertEqual(0, 0)

def tearDown(self):
print('tearDown')

@classmethod
def tearDownClass(cls):
print('tearDownClass')


class MyTestClassTwo(unittest.TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')

def setUp(self):
print('setUp')

def test_third(self):
print('test_third')
self.assertEqual(0, 0)

def test_fourth(self):
print('test_fourth')
self.assertEqual(0, 0)

def tearDown(self):
print('tearDown')

@classmethod
def tearDownClass(cls):
print('tearDownClass')


if __name__ == '__main__':
unittest.main()

后续我们在其他的Python文件中导入​​testclass.py​​：

import unittest
from learning_unittest import testclass

if __name__ == '__main__':
testcase = unittest.TestLoader().loadTestsFromTestCase(testclass.MyTestClassOne)
suite = unittest.TestSuite([testcase])
unittest.TextTestRunner().run(suite)

运行结果如下：

Ran 2 tests in 0.000s
OK
setUpClass
setUp
test_first
tearDown
setUp
test_second
tearDown
tearDownClass

Process finished with exit code 0

上方代码先将​​testclass​​​文件中的内容通过​​import​​​导入，然后使用​​unittest​​​提供的​​unittest.TestLoader().loadTestsFromTestCase​​​方法传入​​testclass​​​文件中的​​class​​​，即​​MyTestClassOne​​​。接下来通过​​unittest.TestSuite([testcase])​​​组装测试用例集合。最后通过​​unittest.TextTestRunner​​​提供的​​run​​方法来执行组装好的集合。但如果：

# testclass.py
import unittest


class MyTestClassOne(unittest.TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')

def setUp(self):
print('setUp')

def test_first(self):
print('test_first')
self.assertEqual(0, 0)

def test_second(self):
print('test_second')
self.assertEqual(0, 0)

def tearDown(self):
print('tearDown')

@classmethod
def tearDownClass(cls):
print('tearDownClass')


class MyTestClassTwo(unittest.TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')

def setUp(self):
print('setUp')

def test_third(self):
print('test_third')
self.assertEqual(0, 0)

def test_fourth(self):
print('test_fourth')
self.assertEqual(0, 0)

def tearDown(self):
print('tearDown')

@classmethod
def tearDownClass(cls):
print('tearDownClass')


if __name__ == '__main__':
testcase = unittest.TestLoader().loadTestsFromTestCase(MyTestClassOne)
suite = unittest.TestSuite([testcase])
unittest.TextTestRunner().run(suite)

即将组装​​testcase​​语句写入​​testclass​​文件中，则运行​​testclass​​文件会执行​​testclass​​文件中的所有​​class​​，而不是只执行​​MyTestClassOne​​：

============================= test session starts =============================
platform win32 -- Python 3.6.4, pytest-3.3.2, py-1.5.2, pluggy-0.6.0
rootdir: D:\LearningPython\learning_unittest, inifile:
collected 4 items
testclass.py setUpClass
.setUp
test_first
tearDown
.setUp
test_second
tearDown
tearDownClass
setUpClass
.setUp
test_fourth
tearDown
.setUp
test_third
tearDown
tearDownClass


========================== 4 passed in 0.03 seconds ===========================
Process finished with exit code 0

通过​​addTest​​构建测试集合

我们可以通过​​addTest​​​将某个​​class​​下面的测试用例添加到集合，然后执行测试集合：

import unittest
from learning_unittest import testclass

if __name__ == '__main__':
suite = unittest.TestSuite()
suite.addTest(testclass.MyTestClassOne('test_first'))
suite.addTest(testclass.MyTestClassTwo('test_third'))
unittest.TextTestRunner().run(suite)

上述代码从​​MyTestClassOne​​​类中取了​​test_first​​​测试用例，从​​MyTestClassTwo​​​类中取了​​test_third​​测试用例，两者组成了一个测试用例集合，运行结果如下：

Ran 2 tests in 0.000s
OK
setUpClass
setUp
test_first
tearDown
tearDownClass
setUpClass
setUp
test_third
tearDown
tearDownClass

Process finished with exit code 0

通过​​discover​​构建测试集合

我们还可以通过​​unittest.TestLoader().discover('.')​​​在指定目录中寻找符合条件的测试用例，从而组成测试集合。关于​​discover​​的用法请参考代码：

import unittest
from learning_unittest import testclass

if __name__ == '__main__':
suite = unittest.TestLoader().discover('.')
unittest.TextTestRunner(verbosity=2).run(suite)

这里的​​discover​​​，我们传递的目录是一个​​‘.’​​​，代表文件所在目录。执行该文件的话，就会从该文件所在目录中去寻找所有符合条件的测试用例。​​discover​​的用法我们会在后续的文章再次讲解。