系统学习Python——单元测试unittest：执行测试用例

分类目录：《系统学习Python》总目录
相关文章：
· 单元测试unittest：框架结构
· 单元测试unittest：测试固件
· 单元测试unittest：编写测试用例
· 单元测试unittest：执行测试用例
· 单元测试unittest：用例执行次序
· 单元测试unittest：内置装饰器
· 单元测试unittest：命令行执行测试
· 单元测试unittest：批量执行测试文件
· 单元测试unittest：测试断言
· 单元测试unittest：测试报告

unittest

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

脚本自测

unittest.main

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

import unittest

blog_name ='hy592070616'classMyTestClass(unittest.TestCase):@classmethoddefsetUpClass(cls):print('setUpClass')defsetUp(self):
        self.blog_name ='hy592070616'print('setUp')deftest_first(self):
        blog_name_ref = blog_name +'--machine learning'
        self.assertEqual(blog_name_ref,'hy592070616--machine learning')deftearDown(self):print('tearDown')@classmethoddeftearDownClass(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.pyimport unittest

classMyTestClassOne(unittest.TestCase):@classmethoddefsetUpClass(cls):print('setUpClass')defsetUp(self):print('setUp')deftest_first(self):print('test_first')
        self.assertEqual(0,0)deftest_second(self):print('test_second')
        self.assertEqual(0,0)deftearDown(self):print('tearDown')@classmethoddeftearDownClass(cls):print('tearDownClass')classMyTestClassTwo(unittest.TestCase):@classmethoddefsetUpClass(cls):print('setUpClass')defsetUp(self):print('setUp')deftest_third(self):print('test_third')
        self.assertEqual(0,0)deftest_fourth(self):print('test_fourth')
        self.assertEqual(0,0)deftearDown(self):print('tearDown')@classmethoddeftearDownClass(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 in0.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.pyimport unittest

classMyTestClassOne(unittest.TestCase):@classmethoddefsetUpClass(cls):print('setUpClass')defsetUp(self):print('setUp')deftest_first(self):print('test_first')
        self.assertEqual(0,0)deftest_second(self):print('test_second')
        self.assertEqual(0,0)deftearDown(self):print('tearDown')@classmethoddeftearDownClass(cls):print('tearDownClass')classMyTestClassTwo(unittest.TestCase):@classmethoddefsetUpClass(cls):print('setUpClass')defsetUp(self):print('setUp')deftest_third(self):print('test_third')
        self.assertEqual(0,0)deftest_fourth(self):print('test_fourth')
        self.assertEqual(0,0)deftearDown(self):print('tearDown')@classmethoddeftearDownClass(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 in0.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 in0.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

的用法我们会在后续的文章再次讲解。