前言
在本章中,你讲学习如何使用Python模块unittest中的工具来测试代码。你将学习编写测试用例,核实一系列输入都将得到预期的输出。你将看到测试通过了是什么样子,测试未通过是什么样子,还将知道测试未通过如何有助于改进代码。你将学习如何测试函数和类,并将知道该为项目编写多少个测试。
1. 单元测试:核实函数在某方面没有问题
2. 测试用例:一组单元测试(涵盖各种函数输入)
3. 全覆盖式测试用例:一整套单元测试(涵盖函数各种使用凡是)
4. 测试函数的步骤:
- (1) 导入unittest模块和要测试的函数
- (2) 创建测试类,并继承unittest.TestCase类
- (3) 定义"test_"打头的测试方法,先调用要测试的函数,再使用断言方法assert()
- (4) unittest.main,让python运行这个文件中的测试
注:运行test_name_function.py时,所有test_打头的方法,自动运行
name_function.py
def get_formatted_name(first, last):
"""Generate a neatly formatted full name."""
full_name = first + ' ' + last
return full_name.title()
test_name_function.py
import unittest
from name_function import get_formatted_name
class NamesTestCase(unittest.TestCase):
def test_first_last_name(self):
"""能正确的处理像Janis Joplin这样的名字吗?"""
formatted_name = get_formatted_name('janis', 'joplin')
self.assertEqual(formatted_name, "Janis Joplin")
if __name__ == '__main__':
unittest.main()
5. 断言方法:检查你认为应该满足的条件是否确实满足。预期和实际
- assertEqual(a, b)
- assertNotEqual(a, b)
- assertTrue(x)
- assertFalse(x)
- assertIn(item, list)
- assertNotIn(item, list)
6. 测试类的步骤
(1) 导入unittest模块和要测试的类
(2) 创建测试类,并继承unittest.TestCase类
(3) "test_"打头的测试方法,先创建一个要测试的类的对象,再调用类中要测试的函数,然后调用断言方法。
(4) unittest.main
survey.py
class AnonymousSurvey():
"""收集匿名调查问卷的答案"""
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)
test_survey.py
import unittest
from survey import AnonymousSurvey
class TestAnonymousSurvey(unittest.TestCase):
"""针对AnonymousSurvey类的测试"""
def test_store_single_response(self):
"""测试单个答案会被妥善存储"""
question = "What language did you first learn to speak?"
my_survey = AnonymousSurvey(question)
my_survey.store_response('English')
self.assertIn('English', my_survey.responses)
if __name__ == "__main__":
unittest.main
7. setUp()
unittest.TestCase类包含setUp()方法,我们只需在setUp()方法中创建一次对象,就可以在每个测试方法中使用它。
python首先运行setUp()方法,再运行各个以“test_”打头的方法
import unittest
from survey import AnonymousSurvey
class TestAnonymousSurvey(unittest.TestCase):
"""针对AnonymousSurvey类的测试"""
def setUp(self):
"""创建一个调查对象和一组答案,供测试方法使用"""
question = "What language did you first learn to speak?"
self.my_survey = AnonymousSurvey(question)
self.responses = ['English', 'Spanish', 'Mandarin']
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_response(self):
"""测试三个答案会被妥善地存储"""
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()