python的unittest模块

狼啸风云

修改于 2022-09-03 20:58:13

6320

修改于 2022-09-03 20:58:13

文章被收录于专栏：计算机视觉理论及其实现计算机视觉理论及其实现

unittest 原名为 PyUnit，是由 java 的 JUnit 衍生而来

unittest库提供了test cases, test suites, test fixtures，test runner。

test cases 通过继承TestCase类，创建一个test或者多个test
test suites 测试套件，相当于test的容器
test fixtures 通常为setup + test case + teardown结构
test runner 执行测试用例tests或suit，可以指定执行方式

整体结构

待测试的类：

class Count:

def __init__(self,a,b):

self.a = a

self.b = b

def add(self):

return self.a+self.b

示例1：

import unittest

class TestDemo(unittest.TestCase):


   def setUp(self):

       print("setUp")


   def test_add(self):

       z = Count(2,3)

       self.assertEqual(z.add(),5)


   def test_invid_add(self):

       z = Count(2, 3)
 
       self.assertNotEqual(z.add(),6)


   def tearDown(self):

       print("tearDown")


if __name__=="__main__":

   suit=unittest.TestSuite()#在pycharm中，手动修改以一般python文件方式去运行

   suit.addTest(TestDemo("test_invid_add"))

   suit.addTest(TestDemo("test_add"))

   runner=unittest.TextTestRunner()

   runner.run(suit)

   # unittest.main() #默认按照字母和数字顺序执行,等同于unittest.makeSuite(test)

示例2：

import unittest

lass TestDemo(unittest.TestCase):

    @classmethod

    def setUpClass(cls):

      print("setUpClass")


    def test_add(self):

      print("test_add")

      z = Count(2,3)

      self.assertEqual(z.add(),5)


    def test_invid_add(self):

      print("test_invid_add")

      z = Count(2, 3)

      self.assertNotEqual(z.add(),6)


    @classmethod

    def tearDownClass(cls):

       print("tearDownClass")



if __name__=="__main__":

   suit=unittest.TestSuite()

   suit.addTest(TestDemo("test_time"))

   suit.addTest(TestDemo("test_add"))

   suit.addTest(TestDemo("test_invid_add"))

   runner=unittest.TextTestRunner()

   runner.run(suit)

从以上2个示例可知道：

unittest默认是按照字母和数字的顺序运行，倘若需要按照我们指定的顺序执行，可使用 suit.addTest()的方式去指定
setUp()、tearDown()会在每个测试用例(test开头)之前和之后执行
setUpClass(cls)、tearDownClass(cls)仅在用例之前和之后各执行1次，与上述的setUp()、tearDown()不同
使用pycharm IDE运行unittest文件，不管是默认的用例执行顺序，还是指定的用例执行顺序，用例执行结果均为默认的顺序，即按照字母和数字的顺序执行。这算是pycharm的一个小小bug，具体做法：进入运行配置界面，在Python 下手动创建指定执行的py文件，而不是Python tests下

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2019-11-04 ，如有侵权请联系 cloudcommunity@tencent.com 删除

uml