import unittest
from survey import AnonymousSurvey
class TestAnonymouSurvey(unittest.TestCase):
    def setUp(self):
        question = "What language did you first learn to speak?"
        self.my_survey = AnonymousSurvey(question)
        self.responses = ['English','Chinese','Franch']
    def test_store_single_response(self):
        self.my_survey.store_responses('English')
        self.assertIn('English',self.my_survey.responses)
    def test_store_three_responses(self):
        for response in self.responses:
            self.my_survey.store_responses(response)
        for response in self.responses:
            self.assertIn(response,self.my_survey.responses)
unittest.main()
