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_storeSingle_response(self): self.my_survey.save_response(self.responses[0]) self.assertIn('English', self.my_survey.responses) def test_storeThree_responses(self): for response in self.responses: self.my_survey.save_response(response) for response in self.responses: self.assertIn(response, self.my_survey.responses)
from survey import AnonymousSurvey question = 'What language did you first learn to speak?' mySurvey = AnonymousSurvey(question) mySurvey.show_question() print("Enter 'q' at any time to quit.\n") while True: response = input("Language: ") if response == 'q': break mySurvey.save_response(response) print("\n Thank you for your participation!\n") mySurvey.show_results()