Python Unittest原理及基本使用方法
基本使用步骤：
1、import unittest
2、新建一个Class继承unittest.TestCase
3、定义一个以test开头的函数作为测试用例【必须以test开头】
4、执行测试用例前都会先调用Setup函数【每个测试用例调用一次】或者setupClass函数【所有测试用例仅调用一次】
5、执行完测试用例后都会调用tearDown函数【多次】和tearDownClass函数【仅一次】
PS：
1、不能重写init()
2、函数必须以test开头才能作为测试用例
3、setupClass()和tearDownClass()必须用@classmethod修饰
# -*- encoding=utf-8 -*-
import unittest
class Demo1(unittest.TestCase):
def setUp(self):
print '调用 setup 函数'
@classmethod
def setUpClass(cls):
print '调用 setupClass 函数'
def tearDown(self):
print '调用 tearDown 函数'
@classmethod
def tearDownClass(cls):
print '调用 tearDownClass 函数'
@staticmethod
def test_add():
print '3 + 5 = {}'.format(3 + 5)
@staticmethod
def test_sub():
print '3 - 5 = {}'.format(3 - 5)
if __name__ == '__main__':
unittest.main()
运行结果
调用 setupClass 函数
调用 setup 函数
3 + 5 = 8
调用 tearDown 函数
调用 setup 函数
3 - 5 = -2
调用 tearDown 函数
调用 tearDownClass 函数
Ran 2 tests in 0.012s
OK
TestSuite【测试组】添加测试用例进行测试
# -*- encoding=utf-8 -*-
class Class1(object):
def __init__(self):
print '调用 Class1 初始化函数'
self.name = '未定义姓名'
self.age = '未定义年龄'
def set_name(self, name):
self.name = name
def set_age(self, age):
self.age = age
def print_name(self):
print 'Class1 的姓名是：{}'.format(self.name)
def print_age(self):
print 'Class1 的年龄是:{}'.format(self.age)
if __name__ == '__main__':
pass
# -*- encoding=utf-8 -*-
import unittest
from Class1 import Class1
class CallClass1(unittest.TestCase):
def setUp(self):
self.class1 = Class1()
pass
def tearDown(self):
self.class1 = None
pass
def test_set_name(self):
self.class1.set_name('设置了名字')
self.class1.print_name()
def test_print_name(self):
self.class1.print_name()
if __name__ == '__main__':
unittest.main()
# -*- encoding=utf-8 -*-
import unittest
from Test_Demo2 import CallClass1
if __name__ == '__main__':
testSuite = unittest.TestSuite()
testSuite.addTest(CallClass1('test_set_name'))
testSuite.addTests(
[CallClass1('test_set_name'), CallClass1('test_print_name')])
textTestRunner = unittest.TextTestRunner()
textTestRunner.run(testSuite)
运行结果
调用 Class1 初始化函数
Class1 的姓名是：设置了名字
调用 Class1 初始化函数
Class1 的姓名是：设置了名字
调用 Class1 初始化函数
Class1 的姓名是：未定义姓名
...
----------------------------------------------------------------------
Ran 3 tests in 0.000s
OK
Process finished with exit code 0
记录测试报告到txt文件
# -*- encoding=utf-8 -*-
import unittest
from Test_Demo2 import CallClass1
if __name__ == '__main__':
testSuite = unittest.TestSuite()
testSuite.addTest(CallClass1('test_set_name'))
testSuite.addTests(
[CallClass1('test_set_name'), CallClass1('test_print_name')])
# textTestRunner = unittest.TextTestRunner()
# textTestRunner.run(testSuite)
# 将测试结果输出到测试报告中
with open('UnittestTextReport.txt', 'w') as f:
runner = unittest.TextTestRunner(stream=f, verbosity=2)
runner.run(testSuite)
记录测试报告到html文件
# 将测试结果输出到测试报告html中
with open('HTMLReport.html', 'w') as f:
runner = HTMLTestRunner(stream=f,
title='MathFunc Test Report',
description='generated by HTMLTestRunner.',
verbosity=2
)
runner.run(suite)
断言
当断言为真时，会执行断言之后的代码如（在断言之后打印成功提示信息）
若断言为假，则之后的代码不会被执行，打应错误信息，之后提示信息不会被执行；unittest中提示F，失败，若捕获到异常，则为E，代表错误；
补充：若断言为假，则当前函数def，停止运行，但会运行之后的def函数
重点：try语句块中不要使用assert一起使用，assert断言为假时，会抛出异常，测试用例为失败，如果放到try中，则会捕获到异常，测试用例永远标记为通过
断言结果：
. 表示通过 或者 pass
F False, 表示断言没有通过
E Error, 表示程序内部发生了错误。
查找当前目录下所有的测试用例【默认以test开头的py文件，可以自己配置】
# -*- encoding=utf-8 -*- import unittest import os if __name__ == '__main__': loader = unittest.defaultTestLoader.discover(os.getcwd()) # tests = unittest.defaultTestLoader.discover( # start_dir='.', # => 指定查找的目录 # pattern='test*.py' # => 指定查找规则 # ) runner = unittest.TextTestRunner() runner.run(loader)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。
下一篇：python如何编写类似nmap的扫描工具