python中使用 unittest.TestCase單元測(cè)試的用例詳解

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

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

import unittest
from unittest import TestCase

from name_function import get_formatted_name


class NamesTestCase(TestCase):
    '''測(cè)試name_function.py'''

    def test_first_last_name(self):
        '''能夠正確地處理象 Janis Joplin這樣的姓名嗎？'''
        formtted_name = get_formatted_name('janis', 'joplin')
        self.assertEqual(formtted_name, 'Janis Joplin')


# 執(zhí)行
unittest.main()

python test_name_function.py

class AnonmousSurvey():
    """收集匿名調(diào)查問卷的答案"""

    def __init__(self, question):
        """存儲(chǔ)一個(gè)問題，并為存儲(chǔ)答案做準(zhǔn)備"""
        self.question = question
        self.responses = []

    def show_question(self):
        """顯示調(diào)查問卷"""
        print(self.question)

    def store_response(self, new_response):
        """存儲(chǔ)單份調(diào)查答卷"""
        self.responses.append(new_response)

    def show_results(self):
        """顯示收集到的所有答卷"""
        print("Survey results")
        for response in self.responses:
            print('- ' + response)

from survey import AnonmousSurvey

# 定義一個(gè)問題，并創(chuàng)建一個(gè)表示調(diào)查的AnonymousSurvey對(duì)象
question = "What language did you first learn to speak?"
my_survey = AnonmousSurvey(question)

# 顯示問題并存儲(chǔ)答案
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)

# 顯示調(diào)查結(jié)果
print("\nThank you to everyoune who participated in the survey!")
my_survey.show_results()

import unittest

from survey import AnonmousSurvey


class TestAnonmousSurvey(unittest.TestCase):
    """針對(duì)AnonymousSurvey類的測(cè)試"""

    def test_store_single_response(self):
        """測(cè)試單個(gè)答案會(huì)被妥善地存儲(chǔ)"""
        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):
        """測(cè)試多個(gè)答案是否會(huì)被存儲(chǔ)"""
        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()

import unittest

from survey import AnonmousSurvey


class TestAnonmousSurvey(unittest.TestCase):
    """針對(duì)AnonymousSurvey類的測(cè)試"""

    def setUp(self):
        """創(chuàng)建一個(gè)調(diào)查對(duì)象和一組答案，供使用的測(cè)試方法使用"""
        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):
        """測(cè)試單個(gè)答案會(huì)被妥善地存儲(chǔ)"""
        self.my_survey.store_response(self.responses[0])
        self.assertIn(self.responses[0], self.my_survey.responses)

    def test_store_three_responses(self):
        """測(cè)試多個(gè)答案是否會(huì)被存儲(chǔ)"""
        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()