Unittest

刚刚学习了Unittest模块，这里简单总结下

函数测试
可通过测试
先创建一个要测试的代码，name_function.py

# -*- coding:utf-8 -*-

def get_formatted_name(first, last):
    '''
    生成一个整洁的格式化全名
    1、在名和姓之间加上一个空格
    2、返回的全名首字母大写
    '''
    full_name = first + ' ' + last
    return full_name.title()

接下来，我们创建一个用来检查get_formatted_name函数在给出名和姓时，能否正确地工作

1. 这里我们创建了一个叫做NamesTestCase的类，它继承自unitest.TestCase
2. 方法名必须以test_打头
3. 给get_formatted_name()方法添加了两个参数‘jiahao’ 'shen'
4. 使用assertEqual断言比较formatted_name变量和后面字符串，如果相等name_function.py代码就没问题，如果不想等就报错！

# -*- coding:utf-8 -*-

import unittest
from .name_function import get_formatted_name


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

    def test_first_last_name(self):
        '''
        能够正确地处理Jiahao Shen这样的姓名么？
        '''
        formatted_name = get_formatted_name('jiahao', 'shen')
        self.assertEqual(formatted_name, 'Jiahao Shen')


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

.表示有一个测试通过了，接下来一行指出python运行了一个测试，消耗时间不到0.003秒

image

不可通过测试

这里修改下get_formatted_name函数，使其能够处理中间名

# -*- coding:utf-8 -*-

def get_formatted_name(first, middle, last):
    '''
    生成一个整洁的格式化全名
    1、在名和姓之间加上一个空格
    2、返回的全名首字母大写
    '''
    full_name = first + ' ' + middle + ' ' + last
    return full_name.title()

再次测试代码会出现报错，它告诉你还少了一个位置参数

image

测试通过了意味着函数的行为是对的，测试未通过意味着新编写的代码有错，因此测试未通过时，不要修改测试，而应该修改代码，这里我们把中间名变成可选的来避免代码错误

修改name_function.py

# -*- coding:utf-8 -*-

def get_formatted_name(first,last, middle=''):
    '''
    生成一个整洁的格式化全名
    1、在名和姓之间加上一个空格
    2、返回的全名首字母大写
    '''
    if middle:
        full_name = first + ' ' + middle + ' ' + last
    else:
        full_name = first + ' ' + last
    return full_name.title()

再次测试，代码测试完成

添加新测试
在测试代码中添加一个新方法，用于添加中间名的测试

# -*- coding:utf-8 -*-

import unittest
from .name_function import get_formatted_name


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

    def test_first_last_name(self):
        '''
        能够正确地处理Jiahao Shen这样的姓名么？
        '''
        formatted_name = get_formatted_name('jiahao', 'shen')
        self.assertEqual(formatted_name, 'Jiahao Shen')

    def test_first_last_middle_name(self):
        '''
        能够正确处理Jiahao Shen Treehl这样的姓名么？
        '''
        formatted_name = get_formatted_name('jiahao', 'shen', 'treehl')
        self.assertEqual(formatted_name, 'Jiahao Treehl Shen')

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

再次测试，两个测试都通过了！！！

类测试
先编写一个类做测试，关于帮助管理匿名调查的类

# -*- coding:utf-8 -*-


class AnonymousSurvey():
    '''
    收集匿名调查问卷的答案
    '''

    def __init__(self, question):
        '''
        存储一个问题，并为存储答案做准备
        1、存储一个问题
        2、创建一个空的列表用来存储答案
        '''
        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类
核实单个答案的测试

# -*- coding:utf-8 -*-

import unittest
from .survey import AnonymousSurvey


class TestAnonymousSurvey(unittest.TestCase):
    '''
    针对AnonymousSurevy类的测试
    1、创建测试单个答案的方法test_store_single_response
    2、创建AnonymousSurvey实例my_survey
    3、使用assertIn来检查Chinese是否在答案列表中
    '''

    def test_store_single_response(self):
        '''
        测试单个答案会被妥善保存
        :return:
        '''
        question = "What language did you first learn to speak?"
        my_survey = AnonymousSurvey(question)
        my_survey.store_response('Chinese')
        self.assertIn('Chinese', my_survey.responses)

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

核实多个答案的测试，我们在TestAnonymouseSurvey类中再添加一个方法

    def test_store_three_responses(self):
        '''
        测试三个答案
        1、使用for循环每个答案
        2、再使用for循环来确认每个答案都在列表内
        '''
        question = "What language did you first learn to speak?"
        my_survey = AnonymousSurvey(question)
        responses = ['Chinese', 'English', 'Chongming']
        for response in responses:
            my_survey.store_response(response)
        for response in responses:
            self.assertIn(response, my_survey.responses)

setUp方法

使用setUp()来创建一个调查对象和一组答案，供test_store_single_response()和test_store_three_responses()使用
可以在setUp()方法中创建一系列实例并设置它们的属性，再在测试方法中直接使用这些实例，有点类似于Django中的ClassBaseView让代码更加容易扩展而不是修改代码

# -*- coding:utf-8 -*-

import unittest
from .survey import AnonymousSurvey


class TestAnonymousSurvey(unittest.TestCase):
    '''
    针对AnonymousSurevy类的测试
    1、创建测试单个答案的方法test_store_single_response
    2、创建AnonymousSurvey实例my_survey
    3、使用assertIn来检查Chinese是否在答案列表中
    '''

    def setUp(self):
        '''
        创建一个调查对象和一组答案
        '''
        question = "What language didi you first learn to speak?"
        self.my_survey = AnonymousSurvey(question)
        self.responses = ['Chinese', 'English', 'France']


    def test_store_single_response(self):
        '''
        测试单个答案
        :return:
        '''
        self.my_survey.store_response(self.responses[0])
        self.assertIn(self.responses[0], self.my_survey.responses)

    def test_store_three_responses(self):
        '''
        测试三个答案
        1、使用for循环每个答案
        2、再使用for循环来确认每个答案都在列表内
        '''
        for response in self.responses:
            self.my_survey.store_response(response)
        for response in self.responses:
            self.assertIn(response, self.my_survey.responses)

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

GitHub
欢迎访问博客Treehl的博客