断言使用
-
我们可以直接在测试用例里面添加unittest框架自带的断言(也就相当于判断),断言方法提供了测试结果是True还是False,所有的断言方法都有一个msg参数,如果指定msg参数的值,则将该信息作为失败的错误信息返回;
-
如果断言失败则抛出一个AssertionError,并标识该测试为失败状态;如果成功,则标识该测试为成功状态;如果异常,则当做错误来处理;
先来看一个实例,在一个测试类MyTestCase中添加多个测试方法和断言
from unittest import TestCase, main, skip
class MyTestCase(TestCase):
def setUp(self):
print('setUp')
def tearDown(self):
print('tearDown')
def test_print1(self):
print('第一个测试用例')
# unittest框架自带的断言,如果不相等,直接抛出异常,相等什么都不做
self.assertEqual(200, 200, '测试失败')
def test_print2(self):
print('第二个测试用例')
def test_print3(self):
print('第三个测试用例')
if __name__ == '__main__':
main()
========输出========:
setUp
第一个测试用例
tearDown
setUp
第二个测试用例
tearDown
setUp
第三个测试用例
tearDown
这里有三个测试用例,每个测试用例开始之前都先执行了setUp方法,每个测试用例之后都执行了tearDown方法。如果我们想要忽略某个测试用例不执行,要怎么做?比如忽略第三个测试用例,只需要在第三个测试用例之前加上skip装饰器:
@skip('do not need test_print3')
def test_print3(self):
print('第三个测试用例')
skip装饰器
-
我们可以使用unitest.skip装饰器跳过不需要的测试用例,
@unittest.skip(reason)
表示无条件跳过测试用例,reason是用来说明原因的; -
@unittest.skipIf(conditition,reason)
:condititon为true时跳过测试用例; -
@unittest.skipunless(condition,reason)
:condition为False时跳过测试用例;
之前我们学习了setup方法和setUpClass方法,会在每个测试用例的前后都执行一遍,那如果我们只想要在所有测试用例的前后执行一遍,也就是在整个类中只执行一遍要怎么办呢,我们可以使用setUpClass方法和tearDownClass方法:
from unittest import TestCase, main, skip
class MyTestCase(TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')
@classmethod
def tearDownClass(cls):
print('tearDownClass')
def test_print1(self):
print('第一个测试用例')
# unittest框架自带的断言,如果不相等,直接抛出异常,相等什么都不做
self.assertEqual(200, 200, '测试失败')
def test_print2(self):
print('第二个测试用例')
@skip('do not need test_print3')
def test_print3(self):
print('第三个测试用例')
if __name__ == '__main__':
main()
==========输出===========
setUpClass
第一个测试用例
第二个测试用例
tearDownClass
setUpClass和tearDownClass的使用
-
setup方法和teardown方法在每次执行一个测试用例时都会重新执行一遍,所以要想在整个类中只执行一遍可以用setUpClass和tearDownClass方法;
-
setUpClass
:需要添加@classmethod 装饰器,在一个类中只会在所有测试用例运行前调用一次; -
tearDownClass
:需要添加@classmethod 装饰器,在所有测试用例运行后调用一次;
引入测试用例集,将测试用例与测试集合分开
- 首先可以先在一个文件中写好测试用例模块,然后在另一个文件中导入这个模块,再使用
unittest.TestSuite类
创建测试套件,然后使用这个类的addTest() 方法将需要被测试的测试用例添加到测试套件中,最后使用unittest.TextTestRunner类
中的run()方法来运行suite所组装的测试用例;
- 测试用例模块
from unittest import TestCase
from unittest import skip
from unittest import main as main_method
class MyTestCase(TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')
@classmethod
def tearDownClass(cls):
print('tearDownClass')
def test_print1(self):
print('第一个测试用例')
self.assertEqual(200, 200, '测试失败')
def test_print2(self):
print('第一个测试用例')
@skip('do not need test_print3')
def test_print3(self):
print('第一个测试用例')
if __name__ == '__main__':
main_method()
- 测试集模块
from test1 import MyTestCase
from unittest import TestSuite, TextTestRunner
suite = TestSuite()
# 在测试用例集中添加测试用例, 只添加两个测试的方法
suite.addTest(MyTestCase('test_print1'))
suite.addTest(MyTestCase('test_print3'))
if __name__ == '__main__':
# 创建测试runner,执行测试用例集
runner = TextTestRunner()
runner.run(suite)
=========输出=============
setUpClass
第一个测试用例
tearDownClass
我们运行这段代码,只有第一个测试用例输出,因为我们只将第一个和第三个测试用例加入到测试集中,所有第二个测试用例不会执行,而我们在第三个测试用例前面机上了skip装饰器会跳过此测试用例,所有第三个也不会执行啦。
将待测试函数、测试用例、测试用例集分开
- 将待测试函数、测试用例、测试用例集分别放在不同的文件中,先在测试用例中导入待测试函数,然后使用断言判断函数是否正确,接着在测试用例集中创健测试套件,且将测试用例加入套件里面,最后运行;
- 测试函数
def add(x, y):
return x + y
def sub(x, y):
return x - y
- 测试用例
from unittest import TestCase
from unittest import skip
from unittest import main as main_method
from add_sub import add, sub
class MyTestCase(TestCase):
@classmethod
def setUpClass(cls):
print('setUpClass')
@classmethod
def tearDownClass(cls):
print('tearDownClass')
def test_print1(self):
ret = 1 + 2
result = add(1, 2)
self.assertEqual(result, ret, 'test_print1测试失败')
def test_print2(self):
print('第二个测试用例')
@skip('do not need test_print3')
def test_print3(self):
print('第三个测试用例')
if __name__ == '__main__':
main_method()
- 测试用例集
from test import MyTestCase
from unittest import TestSuite, TextTestRunner
suite = TestSuite()
# 在测试用例集中添加测试用例
suite.addTest(MyTestCase('test_print1'))
suite.addTest(MyTestCase('test_print2'))
suite.addTest(MyTestCase('test_print3'))
if __name__ == '__main__':
# 创建测试runner,执行测试用例集
runner = TextTestRunner()
runner.run(suite)
========输出=========
setUpClass
第一个测试用例
第二个测试用例
tearDownClass