快捷導(dǎo)航

Python unittest模塊用法實(shí)例分析

更新時(shí)間：2018年05月25日 09:25:07 作者：canlynet

這篇文章主要介紹了Python unittest模塊用法,結(jié)合實(shí)例形式分析了unittest模塊功能及相關(guān)函數(shù)使用技巧,需要的朋友可以參考下

本文實(shí)例講述了Python unittest模塊用法。分享給大家供大家參考，具體如下：

python的unittest模塊提供了一個(gè)測(cè)試框架，只要我們寫一個(gè)繼承unittest.TestCase的類，類中用setUp做初始化，用tearDown做清理。

主要用到的函數(shù)有：

failedinfo表示不成立打印信息failedinfo，為可選參數(shù)
self.fail([msg])會(huì)無(wú)條件的導(dǎo)致測(cè)試失敗，不推薦使用。
self.assertEqual(value1, value2, failedinfo) # 斷言value1 == value2
self.assertTrue(表達(dá)式, failedinfo) # 斷言value為真
self.assertFalse(表達(dá)式, failedinfo) # 斷言value為假
# 斷言肯定發(fā)生異常，如果沒(méi)發(fā)生異常，則為測(cè)試失敗。
# 參數(shù)1為異常，參數(shù)二為拋出異常的調(diào)用對(duì)象，剩余參數(shù)為傳遞給可調(diào)用對(duì)象的參數(shù)。
self.assertRaises(ValueError, self.widget.resize, -1, -1)
調(diào)用時(shí)機(jī)的加self,如self.assertEqual(self.seq, range(10))，self.assertTrue(value > 100)

更詳細(xì)的教程見(jiàn)：http://pyunit.sourceforge.net/pyunit_cn.html

Python代碼：

#coding=utf-8
import random
import unittest
class TestSequenceFunctions(unittest.TestCase):
  def setUp(self):
    self.seq = range(10)
  def test_shuffle(self):
    # make sure the shuffled sequence does not lose any elements
    random.shuffle(self.seq)
    self.seq.sort()
    self.assertEqual(self.seq, range(10))
    # should raise an exception for an immutable sequence
    self.assertRaises(TypeError, random.shuffle, (1,2,3))
  def test_choice(self):
    element = random.choice(self.seq)
    self.assertTrue(element in self.seq)
  def test_sample(self):
    with self.assertRaises(ValueError):
      random.sample(self.seq, 20)
    for element in random.sample(self.seq, 5):
      self.assertTrue(element in self.seq)
results_fields = [
  ("username", unicode),
  ("showid", unicode),
  ("total_pv", int),
  ("pubdate", unicode),
  ("tags", list),
  ("showname", unicode),
  ("pg", int),
  ("ext", str),
]
results_fields_map = dict(results_fields)
class TestDictValueFormatFunctions(unittest.TestCase):
  def setUp(self):
    self.results = [{
      "username": u"瘋狂豆花",
      "showid": u"130e28f0fe0811e0a046",
      "total_pv": 14503214,
      "pubdate": u"2012-07-07 01:22:47",
      "tags": [
        "軒轅劍",
        "天之痕"
        ],
      "showname" : u"軒轅劍之天之痕",
      "pg" : 1,
      "ext" : "mp4"
    }
    ]
  def test_format(self):
    self.assertTrue(isinstance(self.results, list), "self.results's type must be dict but got {0}".format(type(self.results)))
    for r in self.results:
      for f in results_fields_map:
        value = r.get(f, None)
        self.assertTrue(isinstance(value, results_fields_map[f]), u"{0}'s type must be {1} but got {2}".format(value, results_fields_map[f], type(value)))
        #self.assertTrue(isinstance(value, results_fields_map[f]))
  def test_value(self):
    for r in self.results:
      self.assertEqual(r["pg"], 1)
      self.assertEqual(r["ext"], u"mp4")
if __name__ == '__main__':
  # unittest.main() # 用這個(gè)是最簡(jiǎn)單的，下面的用法可以同時(shí)測(cè)試多個(gè)類
  # unittest.TextTestRunner(verbosity=2).run(suite1) # 這個(gè)等價(jià)于上述但可設(shè)置verbosity=2，省去了運(yùn)行時(shí)加-v
  suite1 = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
  suite2 = unittest.TestLoader().loadTestsFromTestCase(TestDictValueFormatFunctions)
  suite = unittest.TestSuite([suite1, suite2])
  unittest.TextTestRunner(verbosity=2).run(suite)

運(yùn)行結(jié)果：

test_choice (__main__.TestSequenceFunctions) ... ok
test_sample (__main__.TestSequenceFunctions) ... ok
test_shuffle (__main__.TestSequenceFunctions) ... ok
test_format (__main__.TestDictValueFormatFunctions) ... ok
test_value (__main__.TestDictValueFormatFunctions) ... ok

----------------------------------------------------------------------
Ran 5 tests in 0.013s

OK

更多Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python入門與進(jìn)階經(jīng)典教程》、《Python字符串操作技巧匯總》、《Python列表(list)操作技巧總結(jié)》、《Python編碼操作技巧總結(jié)》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》及《Python文件與目錄操作技巧匯總》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: