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函数在给出名和姓时，能否正确地工作
- 这里我们创建了一个叫做NamesTestCase的类，它继承自unitest.TestCase
- 方法名必须以test_打头
- 给get_formatted_name()方法添加了两个参数‘jiahao’ 'shen'
- 使用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()
再次测试，两个测试都通过了！！！
| 方法 | 用途 |
|---|---|
| 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中 |
类测试
先编写一个类做测试，关于帮助管理匿名调查的类
# -*- 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()
