常见的python的unittest用法

最新推荐文章于 2025-07-04 19:42:48 发布

转载最新推荐文章于 2025-07-04 19:42:48 发布 · 4.3k 阅读

python 专栏收录该内容

17 篇文章

订阅专栏

本文介绍 Python 的 unittest 框架的使用方法，包括如何编写测试用例、设置测试环境及清理工作，还提供了多种测试场景的示例代码，并介绍了如何整合多个测试文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python的unittest好处是通过python脚本编写用例，每个用例可以单独调试初始化和清理动作，因为都是用例都是代码所以调试起来也很方便；它的缺点是得先学会python，难易程度见仁见智吧，对于我来说那些用excel来写测试用例的框架，我看到里面的用例头都大，还不如看代码来的简单呢。

#!/usr/bin/env
 python

#
 -*- coding: utf-8 -*-

import os,sys

import time

import datetime

import unittest

class NginxTest(unittest.TestCase):

    def setUp(self): #如果不需要每个case都预置和清理环境，而是每个class用一次，只需要用setUpClass、tearDownClass代替即可，如果是整个文件只需要用一次，则用要用
 setUpModule() 和 tearDownModule() 这两个函数了，注意是函数，与 TestCase 类同级

        #预置环境

        print '--------------NginxTestSetUp--------------\n'

    def tearDown(self): 

        #清理环境

        print '--------------NginxTestClear--------------\n' 

    def test_nginx(self):

        print 'test_nginx'       

    def test_nginxlog(self):              

        print 'test_nginxlog'

    @unittest.skip("must
 skipping") #必须跳过下面用例，相当少用

    def test_mustskip(self):

        print 'test_mustskip'

    #
 def test_1(self):

        #
 a=1

        #
 return 1

    @unittest.skipUnless(sys.platform.startswith("win"), "requires
 Windows")#根据条件跳过下面这个用例

    def test_maybeskip(self):

        print 'test_maybeskip'

    def suite_1(self):#非test开头的用例在NginxTest中不会被跑到

        print 'suite_1'

    def suite_2(self):

        print 'suite_2'        

class PhpTest(unittest.TestCase):#因为每个接口的预置环境可能不一样，所以每个接口的用例应该都用单独class来包含，不过每个class的用例都还是要用test开头

    def setUp(self):

        #预置环境

        print '--------------PhpTestSetUp--------------\n'

    def tearDown(self): 

        #清理环境

        print '--------------PhpTestClear--------------\n' 

    def test_php(self):

        print 'test_php'       

    def test_phplog(self):              

        print 'test_phplog'

def suite():#这个表示测试集，不要放在class内，否则会提示"没有这样的测试方法在<class'myapp.tests.SessionTestCase'>：的runTest
 "，我觉得它唯一的好处就是调试的时候可以单独调试某个class而已，我一般不用它，调试时可以注释不需要的class啊 ;-)。不同接口用不同的class也是一种用法，不过那样用我下面说的import不同py的方法更好，因为所有用例写在一起的话文件太大了 ;-)。

    suite = unittest.TestSuite()

    suite.addTest(NginxTest("suite_1"))

    suite.addTest(NginxTest("suite_2"))

    suite.addTest(PhpTest("test_php"))

    suite.addTest(PhpTest("test_phplog"))

    unittest.TextTestRunner().run(suite)

if __name__ == '__main__':

    #
 unittest.main(exit = False,verbosity=2)#它是全局方法，把它屏蔽后，不在suite的用例就不会跑，exit = False表示中间有用例失败也继续执行；还有比较常用的verbosity=2，表示显示def名字

    suite()#执行suite

　　如果接口相当多，为了方便维护，建议每个case用独立的py来写，然后用一个“总入口”去import所有py，然后再调用就行了，这是suite就派上用场了

　　举个例子，luatestcase.py如下：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import time
import datetime
import unittest

class LuaTest(unittest.TestCase):
    def runTest(self):
        print 'anything'
    def setUp(self): 
        #预置环境
        print '--------------LuaTestsetUp--------------\n'
    def tearDown(self):  
        #清理环境
        print '--------------LuaTestclear--------------\n'  
        
    def test_lua(self):
        print 'test_lua'        
   
    def test_lualog(self):               
        print 'test_lualog'
        
def casesuite():
    suite = unittest.TestSuite()
    suite.addTest(LuaTest("test_lua"))
    suite.addTest(LuaTest("test_lualog"))
    unittest.TextTestRunner().run(suite)

　　调用luatestcase.py的“总入口py”就得这样写：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os,sys
import time
import datetime
import unittest

import luatestcase

if __name__ == '__main__':
    luatestcase.casesuite()

　　如果luatestcase.py没有用casesuite“收集”它的用例的话，总入口调不了里面的用例的，不知道是不是我的用法有问题，如果有朋友知道欢迎指点。有的文章说用discover可以实现同样的需求，不过我不会。。。

　　如果要保存unitest的测试输出日志，则需要用到TextTestRunner，例子如下

log_file = "log_file.txt"

f = open(log_file, "w")

runner = unittest.TextTestRunner(stream=f,verbosity=2)

unittest.main(exit = False,testRunner=runner)

f.close()