python3.X 简单的单元测试

首先我们得有需要待测试的函数，如下例子：
survey.py

class a():

    def __init__(self,question):
        self.question = question
        self.response = []

    def show_question(self):
        print(self.question)

    def store_response(self, new_response):
        self.response.append(new_response)

    def show_response(self):
        for response in self.response:
            print(response)

然后就是我们的测试代码，如下：

import unittest
from survey import a


class b(unittest.TestCase):

    def setUp(self):                  # U大写
        question = "whst you like"
        self.my_survey = a(question)
        self.response = ["dog", "cat", "pig"]

    def test_one_response(self):
        self.my_survey.store_response(self.response[0])
        self.assertIn(self.response[0], self.my_survey.response) #断言前者包含于后者则通过

最后的结果应该是：

Ran 1 test in 0.008s

OK

重点：setUp()函数的作用是创建一次对象可以为后面多条测试用例使用，当然这里我只写了一个测试用例。