当前位置 博文首页 > python中使用 unittest.TestCase单元测试的用例详解
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中 |
name_function.py
def get_formatted_name(first, last):
'''生成整洁的姓名'''
full_name = first + ' ' + last
return full_name.title()
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 + '.')
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表示改测试用例中的所有测试单元都通过了。
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)
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()
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()方法中创建一系列实例并设置它们的属性，再在测试方法中直接使用这些实例。相比于在每个测试方法中都创建实例并设置属性，这要容易的多。
jsjbwy运行测试用例时，每完成一个单元测试，python都打印一个字符: 测试通过时打印一个
句点; 测试引发错误时打印一个E; 测试导致断言失败时打印一个F。这就是运行测试用例时，在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含很多单元测试，需要运行很长时间，就可以通过观察这些结果来获悉有多少个测试通过了。
