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)