编写函数或类时,还可为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。在程序中添加新代码时,你也可以对其进行测试,确认它们不会破坏程序既有的行为。
使用Python模块unittest中的工具来测试代码
pizza.py
def get_formatted_name(first,last):
#获取全部名字
full_name = first+' '+last
return full_name.title()
testpizza.py
import unittest
from pizza import get_formatted_name
class NamesTestCase(unittest.TestCase):
"""测试name_function.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()
- 可先导入模块unittest 以及要测试的函数,
- 再创建一个继承unittest.TestCase的类,
- 编写一系列方法对函数行为的不同方面进行测试。
- 运行testpizza.py时,所有以test打头的方法都将自动运行。
使用了unittest 类最有用的功能之一:一个断言方法,断言方法用来核实得到的结果是否与期望的结果一致。
各类断言
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 中
setUp() 方法
- 如果你在TestCase 类中包含了方法setUp() ,Python将先运行它,再运行各个以test_打头的方法。
- 在你编写的每个测试方法中都可使用在方法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()
- 可先导入模块unittest 以及要测试的类,
- 再创建一个继承unittest.TestCase的类,
- 创建一个调查对象和一组答案,供使用的测试方法使用
- 编写一系列方法对函数行为的不同方面进行测试。
- 运行testpizza.py时,所有以test打头的方法都将自动运行。
- 使用了unittest 类最有用的功能之一:一个断言方法,用来核实得到的结果是否与期望的结果一致。