测试函数
单元测试用于核实函数的某个方面没有问题。
def get_formatted_name(first, last, middle=''):
"""生成整洁的姓名"""
if middle:
full_name = first + ' ' + middle + ' ' + last
else:
full_name = first + ' ' + last
return full_name.title()
单元测试
import unittest
from name_function import get_formatted_name
class NameTestCase(unittest.TestCase):
"""测试name_function.py"""
# unittest框架中,main()方法定义函数时使用的是以"test_xxx"或"testxxx"的格式
def test_first_last_name(self):
"""能够正确处理像Janis Joplin这样的姓名吗"""
formatted_name = get_formatted_name('Janis', 'Joplin')
self.assertEqual(formatted_name, 'Janis Joplin')