python中使用 unittest.TestCase單元測試的用例詳解

Posted on by

單元測試和測試用例

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中

函數測試

 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 進行單元測試的文章就介紹到這瞭,更多相關python單元測試內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：