python unittest教程_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博客。