编写函数或类时,还可为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。测试让你信心满满,深信即便有更多的人使用你的程序,它也能正确地工作。在程序中添加新代码时,你也可以对其进行测试,确认它们不会破坏程序既有的行为。程序员都会犯错,因此每个程序员都必须经常测试其代码,在用户发现问题前找出它们。
Python标准库中的模块unittest提供了代码测试工具。单元测试用于核实函数的某个方面没有问题;测试用例是一组单元测试,这些单元测试一起核实函数在各种情形下的行为都符合要求。良好的测试用例考虑到了函数可能收到的各种输入,包含针对所有这些情形的测试。全覆盖式测试用例包含一整套单元测试,涵盖了各种可能的函数使用方式。对于大型项目,要实现全覆盖可能很难。通常,最初只要针对代码的重要行为编写测试即可,等项目被广泛使用时再考虑全覆盖。
# Test.py
def get_formatted_name(first, last):
full_name = first + ' ' + last
return full_name.title()
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 + '.')
# 输入
import unittest
from Test import get_formatted_name as name
class NamesTestCase(unittest.TestCase):
def test_first_last_name(self):
formatted_name = name('janis', 'joplin')
self.assertEqual(formatted_name, 'Janis Joplin')
unittest.main()
# 输出
Enter 'q' at any time to quit.
Please give me a first name: janis
Please give me a last name: joplin
Neatly formatted name: Janis Joplin.
Please give me a first name: q
.
----------------------------------------------------------------------
Ran 1 test in 0.005s
OK
这里,我们使用了unittest类最有用的功能之一:一个断言方法。断言方法用来核实得到的结果是否与期望的结果一致。在这里,我们知道 get_formatted_name() 应返回这样的姓名,即名和姓的首字母为大写,且它们之间有一个空格,因此我们期望formatted_name 的值为Janis Joplin。为检查是否确实如此,我们调用 unittest 的方法assertEqual(),并向它传递 formatted_name 和’Janis Joplin’。代码行 self.assertEqual(formatted_name, ‘Janis Joplin’) 的意思是说:“将 formatted_name 的值同字符串’Janis Joplin’进行比较,如果它们相等,就通过该测试,如果它们不相等,则该测试失败!”
不能通过的测试
测试未通过时结果是什么样的呢?我们来修改get_formatted_name(),使其能够处理中间名,但这样做时,故意让这个函数无法正确地处理像Janis Joplin这样只有名和姓的姓名。
# Test.py
def get_formatted_name(first, middle, last):
full_name = first + ' ' + middle + ' ' +last
return full_name.title()
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
middle = input("Please give me a mid name: ")
if last == 'q':
break
formatted_name = get_formatted_name(first, middle, last)
print("\tNeatly formatted name: " + formatted_name + '.')
import unittest
from Test import get_formatted_name as name
class NamesTestCase(unittest.TestCase):
def test_first_last_name(self):
formatted_name = name('janis', 'joplin')
self.assertEqual(formatted_name, 'Janis joplin')
unittest.main()
# 输出
E
======================================================================
ERROR: test_first_last_name (__main__.NamesTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "C:/script/Test2.py", line 6, in test_first_last_name
formatted_name = name('janis', 'joplin')
TypeError: get_formatted_name() missing 1 required positional argument: 'last'
----------------------------------------------------------------------
Ran 1 test in 0.006s
FAILED (errors=1)