Python学习——代码测试

单元测试 用于核实函数的某个方面没有问题,测试用例是一组单元测试,这些单元测试一起核实函数在各种情形下的行为都符合要求

  模块unittest提供了代码测试工具

测试函数

  用于测试的类必须继承unittest.TestCase类

  unittest类最有用的功能之一是:一个断言方法。断言方法用于核实得到的结果是否与期望的结果一致

name_function.py

def get_formatted_name(first, last):

   full_name = first + ' ' + last
return full_name.title()

test.py

import unittest

from name_function import get_formatted_name

class NamesTestCase(unittest.TestCase):

def test_first_last_name(self):

formatted_name = get_formatted_name('janis', 'joplin')

self.assertEqual(formatted_name, 'Janis Joplin')

#unittest.main() 使用pycharm不需要加这句

测试类

  unittest.TestCase类包含方法setUp( ), 可以让我们只需要创建测试中使用的对象一次,并在每个测试方法中使用它们。如果在TestCase中包含了方法setUp( ), Python将

先运行它,再运行各个以test_打头的方法。这样在每个测试方法中就都能使用在方法setUp( )中创建的对象了。

下面是书中测试类的一个例程:

survey.py

class AnonymousSurvey():
def __init__(self, question):
self.question = question
self.responses = []
def store_response(self, new_response):
self.responses.append(new_response)
def show_results(self):
print("Survey results: ")
for response in self.responses:
print('- ' + response)

test_survey.py

import unittest

from survey import AnonymousSurvey

class TestAnonymousSurvey(unittest.TestCase):

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('English',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)