本文實(shí)例講述了python3.6編寫的單元測試。分享給大家供大家參考，具體如下：

使用python3.6編寫一個(gè)單元測試demo，例如：對學(xué)生Student類編寫一個(gè)簡單的單元測試。

1、編寫Student類：

            
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
class Student(object):
  def __init__(self,name,score):
    self.name = name
    self.score = score
  def get_grade(self):
    if self.score >= 80 and self.score <= 100:
      return 'A'
    elif self.score >= 60 and self.score <= 79:
      return 'B'
    elif self.score >= 0 and self.score <= 59:
      return 'C'
    else:
      raise ValueError('value is not between 0 and 100')


          

2、編寫一個(gè)測試類TestStudent，從unittest.TestCase繼承：

            
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import unittest
from student import Student
class TestStudent(unittest.TestCase):
  def test_80_to_100(self):
    s1 = Student('Bart',80)
    s2 = Student('Lisa',100)
    self.assertEqual(s1.get_grade(),'A')
    self.assertEqual(s2.get_grade(),'A')
  def test_60_to_80(self):
    s1 = Student('Bart',60)
    s2 = Student('Lisa',79)
    self.assertEqual(s1.get_grade(),'B')
    self.assertEqual(s2.get_grade(),'B')
  def test_0_to_60(self):
    s1 = Student('Bart',0)
    s2 = Student('Lisa',59)
    self.assertEqual(s1.get_grade(),'C')
    self.assertEqual(s2.get_grade(),'C')
  def test_invalid(self):
    s1 = Student('Bart',-1)
    s2 = Student('Lisa',101)
    with self.assertRaises(ValueError):
      s1.get_grade()
    with self.assertRaises(ValueError):
      s2.get_grade()
#運(yùn)行單元測試
if __name__ == '__main__':
  unittest.main()


          

3、運(yùn)行結(jié)果如下：

4、行單元測試另一種方法：在命令行通過參數(shù)-m unittest直接運(yùn)行單元測試，例如：python -m unittest student_test

最后對使用unittest模塊的一些總結(jié)：

1. 編寫單元測試時(shí)，需要編寫一個(gè)測試類，從 unittest.TestCase 繼承
2. 對每一個(gè)類測試都需要編寫一個(gè) test_xxx() 方法
3. 最常用的斷言就是 assertEqual()
4. 另一種重要的斷言就是期待拋出指定類型的Error，eg： with self.assertRaises(KeyError):
5. 另一種方法是在命令行通過參數(shù)-m unittest直接運(yùn)行單元測試：eg： python -m unittest student_test
6. 最簡單的運(yùn)行方式是xx.py的最后加上兩行代碼：

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

更多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文件與目錄操作技巧匯總》

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