Python Unittest自动化单元测试框架详解

本文实例为大家分享了python unittest自动化单元测试框架的具体代码，供大家参考，具体内容如下

1、python 测试框架(本文只涉及 pyunit)

参考地址

2、环境准备

首先确定已经安装有python，之后通过安装pyunit，python版本比较新的已经集成有pyunit(pyunit 提供了一个图形测试界面unittestgui.py)
参考：查看地址

3、代码实例

使用的ide为 pycharm,demo结构如图

1.简单地一个实例

# test002_fail.py


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

import unittest

class test002_fail(unittest.testcase):

  #测试用例前执行
  def setup(self):
    print 'case before'
    pass

  #测试用例后执行
  def teardown(self):
    print 'case after'
    pass

  #测试用例1
  def test_case1(self):
    a = 3
    b = 2
    self.assertequal(a+b,4,'result fail')

  #测试用例2
  def test_case2(self):
    a = 2
    b = 3
    self.assertequal(a*b,7,'result fail')

if __name__ == '__main__':
  unittest.main()

2.当case比较多的时候，可以使用suite来管理多个case,suite 可以看做是一个case 的容器，可以装很多的case

# suitedemo.py
# -*- coding:utf-8 -*-
#一个类，添加测试集
import unittest

class suitedemo(unittest.testcase):


  #测试用例前执行
  def setup(self):
    print 'case before'
    pass

  #测试用例后执行
  def teardown(self):
    print 'case after'
    pass

  def test_case1(self):
    a = 3
    b = 2
    self.assertequal(a+b,5,'result fail')
    print 'case1'

  def test_case2(self):
    a = 2
    b = 3
    self.assertequal(a*b,6,'result fail')
    print 'case2'

#定义一个测试集合，方便添加case

def suite():
  suitetest = unittest.testsuite()
  suitetest.addtest(suitedemo("test_case1"))
  suitetest.addtest(suitedemo("test_case2"))
  return suitetest


#默认运行时通过 suite 运行

if __name__ == '__main__':
  unittest.main(defaulttest='suite')

3.suite 和 case可以分开写在不同的python文件中，这样方便区分case模块和suite管理模块，在测试测试的时候只需要在case模块里面添加case,之后添加到suite模块即可了

case模块

# testcasedemo

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


import unittest


class testcasedemo(unittest.testcase):
  def setup(self):
    pass

  def teardown(self):
    pass

  def test_case1(self):
    print 'test_case1'

  def test_case2(self):
    print 'test_case2'

  def test_case3(self):
    print 'test_case3'

  def test_case4(self):
    print 'test_case4'

接下来为三个suite模块，都是先导入case模块，之后添加suite,再在suite里面添加case，最后执行的时候需要使用testrunner,这个testrunner 相当于suite 的执行器

简单地testsuite:

# testcase_demo1.py

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

#集合写入一个方法，main调用并启动


import unittest
from testcase_demo import testcasedemo


#添加一个测试集合，并添加case

def suite():
  suitetest = unittest.testsuite()
  suitetest.addtest(testcasedemo('test_case1'))
  return suitetest


#指定并启动测试集合,运行集合方法

if __name__ == '__main__':

  runner = unittest.texttestrunner()
  runner.run(suite())

也可以不定义suite方法，直接在main方法中进行case的添加，之后执行

# testcase_demo2.py

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

#集合可以写在main


import unittest
from testcase_demo import testcasedemo


#指定并启动测试集合

if __name__ == '__main__':

  #添加测试集合case,并启动
  suitetest = unittest.testsuite()
  suitetest.addtest(testcasedemo('test_case1'))
  suitetest.addtest(testcasedemo('test_case2'))

  #直接启动集合
  runner = unittest.texttestrunner()
  runner.run(suitetest)

可以把不同的case 添加道不同的suite里面，同样，可以给suite分组，用于区分不同的case,并且，suite 之间还可以包含

# testcase_demo3.py
# -*- coding:utf-8 -*-

#包含集合，多集合

import unittest
from testcase_demo import testcasedemo

#添加不同的合集

def suite1():
  suitetest = unittest.testsuite()
  suitetest.addtest(testcasedemo("test_case1"))
  suitetest.addtest(testcasedemo("test_case2"))
  print 'suite1 运行'
  return suitetest

def suite2():
  suitetest = unittest.testsuite()
  suitetest.addtest(testcasedemo("test_case3"))
  suitetest.addtest(testcasedemo("test_case4"))
  print 'suite2 运行'
  return suitetest

#包含所有的suite

def allsuite():
  alltest = unittest.testsuite((suite1(),suite2()))
  return alltest

#运行的时候，可以根据不同的要求，运行不同的suite,或者全部运行，这样就方便管理每次运行的case

if __name__ == '__main__':
  runner = unittest.texttestrunner()
  runner.run(allsuite())

4.小结

pyunit 是python的一个单元测试框架，使用它可以方便的管理自动化运行的case.
在python 中，每一个方法 只要方法名使用test_ 开头的都被作为是一个用例，都可以看做成一个case,这里介绍了使用suite管理case，并且分组运行case,还有其他很多的功能，请参考： unittest — unit testing framework

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。