天天看點

python 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()      

繼續閱讀