python中如何使用unittest.TestCase單元測試

這篇文章主要介紹了python中如何使用unittest.TestCase單元測試，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

單元測試和測試用例

python標準庫中的模塊unittest提供了代碼測試工具。單元測試用于核實函數的莫個方面沒有問題；測試用例是一組單元測試，這些單元測試一起核實函數在各種情形下的行為都符合要求。良好的測試用例考慮到了函數可能收到的各種輸入，包含針對所有這些情形的測試。全覆蓋測試用例包含一整套單元測試，涵蓋了各種可能的函數使用方式。對于大型項目，要實現全覆蓋可能很難，通常，最初只要針對代碼的重要行為編寫測試即可，等項目被廣泛使用時再考慮全覆蓋。

各種斷言方法

python 在unittest.TestCase 中提高了很多斷言方法。

unittest Module中的斷言方法

函數測試

 1.準備測試函數

name_function.py

def get_formatted_name(first, last):
    '''生成整潔的姓名'''
    full_name = first + ' ' + last
    return full_name.title()

2.編寫一個能使用它的程序

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 + '.')

3.對函數進行單元測試

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表示改測試用例中的所有測試單元都通過了。

類測試

1.準備測試的類

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)

2.編寫一個能使用它的程序

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()

3.對類進行單元測試

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單元測試”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!