测试函数

首先有一个函数 name_function.py，他接受名和姓，并返回一个整洁的姓名。

def get_formatted_name(first, last):
    """Generate a neatly formatted full name"""
    full_name = first + ' ' + last
    return full_name.title()

函数 name.py 接受用户输入的名和姓，并显示一个整洁的姓名。

from learn_test.name_function import get_formatted_name


def get_name():
    print("Enter 'q' at any time to quit.")
    while True:
        first = input("
Please 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("	Neatly formatted name: " + formatted_name + '.')


if __name__ == '__main__':
    get_name()

要为函数编写测试用例，1. 先导入模块unittest以及要测试的函数 2.再创建一个继承unittest.TestCase的类，并编写一系列方法对函数行为的不同方面进行测试

test_name_function.py

import unittest
from .name_function import get_formatted_name


class NamesTestCase(unittest.TestCase):
    """测试 name_function.py"""

    def test_first_last_name(self):
        """能够正确处理像 Jains Joplin这样的姓名吗"""
        formatted_name = get_formatted_name('janis', 'joplin')
        self.assertEqual(formatted_name, 'Janis Joplin')


unittest.main()

首先，我们导入了模块unittest和要测试的函数get_formatted_name()。

我们创建了一个名为NamesTestCase的类，用于包含一系列针对get_formatted_name（）的单元测试。（可以随便给这个类命名，但最好让他看起来与要测试的函数相关，并包含字样Test ； 这个类必须继承 unittest.TestCase类，这样python才知道如何运行你编写的函数。）

NamesTestCase只包含一个方法，用于测试get_formatted_name（）的一个方面。我们将这个方法命名为test_first_last_name（），因为我们要核实的是只有名和姓的姓名能否被正确的格式化。我们运行test_name_function.py时，所有以test_打头的方法都将自动运行。在这个方法中，我们调用了要测试的函数，并存储了要测试的返回值。

最后，我们使用了unittest类最有用的功能之一：一个断言方法。断言方法用来核实得到的结果与期望的结果是否一致。我们调用unittest的方法assertEqual（），并向他传递formatted_name和‘Jains Joplin’。

代码执行unittest.main（）让Python运行这个文件中的测试。运行test_name_function.py时，得到的输出如下：

测试未通过：

def get_formatted_name(first, middle, last):
    """Generate a neatly formatted full name"""
    full_name = first + ' ' + middle + ' ' + last

将 get_formatted_name 函数进行修改，再运行测试函数，就会发现报错了。

 

测试未通过时，不要修改测试，而应修复导致测试不能通过的代码：检查刚才对函数所做的修改，找出导致函数行为不符合预期的修改。

 添加新测试：

在确定get_formatted_name（）能够正确处理简单的姓名后，我们在编写一个测试，用于测试包含中间名的姓名。在NamesTestCase类中在添加一个新的方法。

上面我们编写了针对单个函数的测试，下面学习编写针对类的测试。如果针对类的测试通过了，就能确信对类所做的改进没有意外的破坏其原有的行为。

只能在继承unittest.TestCase的类中使用这些方法。

各种断言方法：

方法 —— 用途

assertEqual(a, b)	核实 a == b
assertNotEqual(a, b)	核实 a != b
assertTrue(x)	核实 x 为 True
assertFalse(x)	核实 x 为 False
assertIn(item, list)	核实 item 在 list 中
assertNotIn(item, list)	核实 item 不在 list 中

首先我们写了一个帮助管理匿名调查的类

class AnonymousSurvey(object):
    """手机匿名调查问卷的答案"""

    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)

为证明 AnonymousSurvey 类能正确工作，我们编写一个使用他的程序 language_survey.py

from learn_test.survey import AnonymousSurvey


# 定义一个问题， 并创建一个表示调查的 AnonymousSurvey 对象
question = 'What language did you first learn to speak?'
my_survey = AnonymousSurvey(question)

# 显示问题并存储答案
my_survey.show_question()
print("Enter 'q' at any time to quit.
")
while True:
    response = input('Language: ')
    if response == 'q':
        break
    my_survey.store_response(response)

# 显示调查结果
print("
Thank you to everyone who participated in the survey")
my_survey.show_results()

测试类：

import unittest
from learn_test.survey import AnonymousSurvey


class TestAnonymousSurvey(unittest.TestCase):
    """针对AnonymousSurvey类的测试"""

    def test_store_single_response(self):
        """测试单个答案会被妥善的存储"""
        question = "What langusage did you first learn to say?"
        my_survey = AnonymousSurvey(question)
        my_survey.store_response('English')

        self.assertIn('English', my_survey.responses)

    def test_store_three_responses(self):
        """测试三个答案会被妥善的存储"""
        question = "What langusage did you first learn to say?"
        my_survey = AnonymousSurvey(question)
        responses = ['English', 'Spanish', 'Chinese']
        for response in responses:
            my_survey.store_response(response)

        for response in responses:
            self.assertIn(response, my_survey.responses)


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

在前面的测试中，我们在每个测试方法中都创建了一个AnonymousSurvey实例，并在每个方法中都创建了答案。unittest.TestCase类包含方法setUp（）,让我们只需创建对象一次，并在每个方法中使用它们。如果在TestCase类中包含了方法 setUp（），python将先运行他，再运行各个以 test_打头的方法。

class TestAnonymousSurvey(unittest.TestCase):
    """针对AnonymousSurvey类的测试"""
    def setUp(self):
        """创建一个调查对象和一组答案，供使用的测试方法使用。"""
        question = "What langusage did you first learn to say?"
        self.my_survey = AnonymousSurvey(question)
        self.responses = ['English', 'Spanish', 'Chinese']

    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)

方法setUp()做了两件事： 创建一个调查对象  self.my_survey； 创建一个答案列表 self.responses。 存储这两样东西的变量名包含前缀self（即存储在属性中），因此可以在这个类中的任何地方使用。