测试函数
单元测试和测试用例
- Python标准库中的模块unittest提供了代码测试工具
-
单元测试用于核实函数的某个方面没有问题，测试用例是一组单元测试；
-
全覆盖式测试用例包含一整套单元测试
测试函数
# name_func.py
def get_formatted_name(first, last):
full_name = first + ' ' + last
return full_name.title()
#test_name_func.py
import unittest
from name_func import get_formatted_name
class NamesTestCase(unittest.TestCase):
"""测试name_func.py"""
def test_first_last_name(self):
"""能够正确地处理像Janis Joplin这样的姓名吗？ """
formatted_name = get_formatted_name('janis', 'joplin')
self.assertEqual(formatted_name, 'Janis Joplin')
unittest.main()
- 首先，导入模块unittest和要测试的函数get_formatted_name()
- 然后创建一个继承unittest.TestCase的类NamesTestCase，用于包含一系列针对测试函数的单元测试
- NamesTestCase包含一个方法test_first_last_name()，当我们运行test_name_func.py时，所有以test_开头的方法都将自动运行。
- 接下来调用unittest的断言方法assertEqual()，用以比较函数结果和正确结果，如果相等，则万事大吉，如果不相等就和我说一声。
- 测试成功，将输出如下
.
----------------------------------------------------------------------
Ran 1 tests in 0.000s
OK
- 运行测试用例时，每完成一个单元测试， Python都打印一个字符：测试通过时打印一个句点；测试引发错误时打印一个E；测试导致断言失败时打印一个F。
添加新测试
import unittest
from name_func import get_formatted_name
class NamesTestCase(unittest.TestCase):
"""测试name_func.py"""
def test_first_last_name(self):
"""能够正确地处理像Janis Joplin这样的姓名吗？ """
formatted_name = get_formatted_name('janis', 'joplin')
self.assertEqual(formatted_name, 'Janis Joplin')
def test_first_last_middle_name(self):
"""能够正确地处理像Wolfgang Amadeus Mozart这样的姓名吗？ """
formatted_name = get_formatted_name('wolfgang', 'mozart', 'amadeus')
self.assertEqual(formatted_name, 'Wolfgang Amadeus Mozart')
unittest.main()
- 同样，新添加的方法必须以test_打头，这个方法测试函数的另一个行为，即处理带中间名的名字。
- 测试成功，将输出如下
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
测试类
各种断言方法
- Python在unittest.TestCase类中提供了很多断言方法，断言方法检查你认为应该满足的条件是否满足。
- 下表描述了6个常用的断言方法
| 方法 |
用途 |
| 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中 |
准备要测试的类
- 类的测试与函数的测试相似，因为你所做的大部分工作都是测试类中方法的行为，但也有一些不同之处。
- 以下是一个要被测试的类
# survey.py
class AnonymousSurvey():
"""收集匿名调查问卷的答案"""
def __init__(self, question):
"""存储一个问题，并为存储答案做准备"""
self.question = question
self.responses = []
def show_question(self):
"""显示调查问卷"""
print(question)
def store_response(self, new_response):
"""存储单份调查答卷"""
self.responses.append(new_response)
def show_results(self):
"""显示收集到的所有答卷"""
print("Survey results:")
for response in responses:
print('- ' + response)
测试AnonymousSurvey类
- 下面编写一个测试，验证：如果用户面对调查问题时只提供了一个答案，这个答案也能被妥善地存储。为此我们用方法assertIn()来核实它包含在答案列表中：
#test_survey.py
import unittest
from survey import AnonymousSurvey
class TestAnonmyousSurvey(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)
unittest.main()
方法setUp()
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_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()做了两件事：创建一个调查对象；创建一个答案列表 。存储这两样东西的变量名包含前缀self，因此可在这个类的任何地方使用。
- 方法 test_store_three_response() 核 实 self.responses 中 的 第 一 个 答 案 self.responses[0] 被妥善地存储 。
- 方 法 test_store_three_response() 核实 self.responses 中的全部三个答案都被妥善地存储。
