這篇文章主要介紹了python中如何使用unittest.TestCase單元測試,具有一定借鑒價值,感興趣的朋友可以參考下,希望大家閱讀完這篇文章之后大有收獲,下面讓小編帶著大家一起了解一下。
python標準庫中的模塊
unittest提供了代碼測試工具。單元測試用于核實函數的莫個方面沒有問題;測試用例是一組單元測試,這些單元測試一起核實函數在各種情形下的行為都符合要求。良好的測試用例考慮到了函數可能收到的各種輸入,包含針對所有這些情形的測試。全覆蓋測試用例包含一整套單元測試,涵蓋了各種可能的函數使用方式。對于大型項目,要實現全覆蓋可能很難,通常,最初只要針對代碼的重要行為編寫測試即可,等項目被廣泛使用時再考慮全覆蓋。
python 在unittest.TestCase 中提高了很多斷言方法。
unittest Module中的斷言方法
| 方法 | 用途 |
|---|---|
| assertEqual(a,b) | 核實a == b |
| assertNotEqual(a,b) | 核實a != b |
| assertTrue(x) | 核實x為True |
| assertFalse(x) | 核實x為False |
| assertIn(item,list) | 核實ietm在list中 |
| assertNotIn(item,list) | 核實item不在list中 |
name_function.py
def get_formatted_name(first, last): '''生成整潔的姓名''' full_name = first + ' ' + last return full_name.title()
nams.py
from name_function import get_formatted_name
print("Enter 'q' at any time to quit.")
while True:
first = input("\nPlease give me a first name: ")
if first == 'q':
break
last = input("Please give me a last name: ")
if last == 'q':
break
formatted_name = get_formatted_name(first, last)
print("\tNeatly formatted name: " + formatted_name + '.')test_name_function.py
import unittest
from unittest import TestCase
from name_function import get_formatted_name
class NamesTestCase(TestCase):
'''測試name_function.py'''
def test_first_last_name(self):
'''能夠正確地處理象 Janis Joplin這樣的姓名嗎?'''
formtted_name = get_formatted_name('janis', 'joplin')
self.assertEqual(formtted_name, 'Janis Joplin')
# 執行
unittest.main()python test_name_function.py

第一行的句點 表示測試通過了,接下來的一行指出python運行了一個測試,消耗的時間不到0.001秒,最后的OK表示改測試用例中的所有測試單元都通過了。
survey.py
class AnonmousSurvey():
"""收集匿名調查問卷的答案"""
def __init__(self, question):
"""存儲一個問題,并為存儲答案做準備"""
self.question = question
self.responses = []
def show_question(self):
"""顯示調查問卷"""
print(self.question)
def store_response(self, new_response):
"""存儲單份調查答卷"""
self.responses.append(new_response)
def show_results(self):
"""顯示收集到的所有答卷"""
print("Survey results")
for response in self.responses:
print('- ' + response)language_survey.py
from survey import AnonmousSurvey
# 定義一個問題,并創建一個表示調查的AnonymousSurvey對象
question = "What language did you first learn to speak?"
my_survey = AnonmousSurvey(question)
# 顯示問題并存儲答案
my_survey.show_question()
print("Enter 'q' at any time to quit.\n")
while True:
response = input("Language: ")
if response == 'q':
break
my_survey.store_response(response)
# 顯示調查結果
print("\nThank you to everyoune who participated in the survey!")
my_survey.show_results()import unittest
from survey import AnonmousSurvey
class TestAnonmousSurvey(unittest.TestCase):
"""針對AnonymousSurvey類的測試"""
def test_store_single_response(self):
"""測試單個答案會被妥善地存儲"""
question = "What language did you first learn to speak?"
my_survey = AnonmousSurvey(question)
my_survey.store_response('English')
self.assertIn('English', my_survey.responses)
def test_store_three_responses(self):
"""測試多個答案是否會被存儲"""
question = "What language did you first learn to speak?"
my_survey = AnonmousSurvey(question)
responses = ["English", "Chinses", "Japan"]
for response in responses:
my_survey.store_response(response)
for response in responses:
self.assertIn(response, my_survey.responses)
unittest.main()
可以看到對類的單元測試也是成功的。雖然成功了,但是做法不是很好,測試有些重復了,下面使用unittest的另一項功能來提高它們的效率
方法 setUP()
如果你在TestCase類中包含方法setUP(),python將先運行它,在運行各個以test_開頭的方法。
test_survey_setup.py
import unittest from survey import AnonmousSurvey class TestAnonmousSurvey(unittest.TestCase): """針對AnonymousSurvey類的測試""" def setUp(self): """創建一個調查對象和一組答案,供使用的測試方法使用""" question = "What language did you first learn to speak?" self.my_survey = AnonmousSurvey(question) self.responses = ["English", "Chinses", "Japan"] def test_store_single_response(self): """測試單個答案會被妥善地存儲""" self.my_survey.store_response(self.responses[0]) self.assertIn(self.responses[0], self.my_survey.responses) def test_store_three_responses(self): """測試多個答案是否會被存儲""" for response in self.responses: self.my_survey.store_response(response) for response in self.responses: self.assertIn(response, self.my_survey.responses) unittest.main()
測試自己編寫的類時,方法setUP()讓測試方法編寫起來更容易:可以在setUP()方法中創建一系列實例并設置它們的屬性,再在測試方法中直接使用這些實例。相比于在每個測試方法中都創建實例并設置屬性,這要容易的多。
運行測試用例時,每完成一個單元測試,python都打印一個字符: 測試通過時打印一個
句點; 測試引發錯誤時打印一個E; 測試導致斷言失敗時打印一個F。這就是運行測試用例時,在輸出的第一行中看到的句點和字符數量各不相同的原因。如果測試用例包含很多單元測試,需要運行很長時間,就可以通過觀察這些結果來獲悉有多少個測試通過了。
感謝你能夠認真閱讀完這篇文章,希望小編分享的“python中如何使用unittest.TestCase單元測試”這篇文章對大家有幫助,同時也希望大家多多支持億速云,關注億速云行業資訊頻道,更多相關知識等著你來學習!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。