| 方法 | 用途 |
|---|---|
| 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()方法中创建一系列实例并设置它们的属性，再在测试方法中直接使用这些实例。相比于在每个测试方法中都创建实例并设置属性，这要容易的多。
运行测试用例时，每完成一个单元测试，python都打印一个字符: 测试通过时打印一个
句点; 测试引发错误时打印一个E; 测试导致断言失败时打印一个F。这就是运行测试用例时，在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含很多单元测试，需要运行很长时间，就可以通过观察这些结果来获悉有多少个测试通过了。
到此这篇关于python中使用 unittest.TestCase 进行单元测试的文章就介绍到这了,更多相关python单元测试内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
咨 询 客 服