Python代码测试
对于编写的代码,可以使用unittest模块中的相关方法进行测试。
测试函数
首先定义一个简单的函数,该函数用来合并名称并返回。
name_function.py:
def get_fromatted_name(first,last):
"""组合姓名并返回"""
full_name=first+' '+last
return full_name.title()
接着编写测试代码,需要先导入unittest模块。然后创建一个测试类,用于包含对上述函数的单元测试,这个类必须继承unittest.TestCase类。
test_name_function.py:
#导入模块和要测试的函数
import unittest
from name_function import get_fromatted_name
#编写测试类,必须继承自unittest.TestCase
class NamesTestCase(unittest.TestCase):
"""测试name_function.py"""
#测试方法
def test_first_last_name(self):
"""能够正确的处理Python Java这样的名字吗?"""
formatted_name=get_fromatted_name('python','java')
#调用断言方法用来核实得到的结果是否与期望值结果一致
self.assertEqual(formatted_name,'Python Java')
#运行测试
unittest.main()
上述中最重要的是调用TestCase类的assertEqual()断言方法,用来核实代码执行结果和预期结果是否一致。
注意:当编写测试类时,除了需要继承unittest.TestCase外,测试类中的测试方法必须以test_开头,这样在调用unittest.main()方法时,这些以test_开头的方法才会自动运行。
unittest.TestCase类的断言方法
断言方法检查你认为应该满足的条件是否确实满足。如果该条件确实满足,你对程序行为的假设就得到了确认,你就可以确信其中没有错误。如果你认为应该满足的条件实际上并不满足,Python将引发异常。
方法
用途
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中
测试类
首先定义一个简单的类,接着对这个类编写测试代码。
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("答案:")
for response in self.responses:
print('-'+response)
测试类和测试函数类似,核心都是应用断言方法来实现的。
import unittest
from survey import AnonymousSurvey
class TestAnonyousSurvey(unittest.TestCase):
"""对类AnonymousSurvey进行测试"""
def setUp(self):
question="你对什么语言感兴趣?"
self.my_survey=AnonymousSurvey(question)
#设定程序回答的答案
self.responses=["java","python","c#"]
#方法以test_开头
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)
unittest.main()
setUp()方法
unittest.TestCase类包含方法setUp(),让我们只需创建这些对象一次,就可以在每个测试方法中使用它们。
如果你在自己的测试类中重写了setUp()方法,python将先运行它,再运行各个以test_开头的方法。这样,在你编写的每个测试方法中都可使用在方法setUp()中创建的对象了。
import unittest
from survey import AnonymousSurvey
class TestAnonyousSurvey(unittest.TestCase):
"""对类AnonymousSurvey进行测试"""
def setUp(self):
question="你对什么语言感兴趣?"
self.my_survey=AnonymousSurvey(question)
#设定程序回答的答案
self.responses=["java","python","c#"]
#方法以test_开头
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)
unittest.main()
总结
python代码单元测试,需要使用unittest模块,尤其是其中的TestCase类及其断言方法。
在编写测试代码时,需要自定义一个测试类,并且继承自unittest.TestCase。
所有编写的测试方法都必须以test_开头,这样在调用unittest.main()方法后,才能自动执行。
如果需要在不同的测试方法中使用同一个对象,可以考虑在类中重写setUp()。
断言方法用于验证程序运行的值和期望值(一般是写死的值)是否一致,从而反馈测试的结果。
参考资源
《Python编程:从入门到实践》
本文后续会随着知识的积累不断补充和更新,内容如有错误,欢迎指正。
最后一次更新时间:2018-08-23