class TestAnonymousSurvey(unittest.TestCase): """Test for class anonymous survey.""" def setUp(self): """ Create a survey and a set of responses for use in all test methods. """ question = 'What language did you first learn to speak?' self.my_survey = AnonymousSurvey(question) self.responses = ['Russian', 'English', 'Chinese'] def test_store_single_rensonse(self): """Test that a single response is stored properly.""" self.my_survey.store_responce(self.responses[0]) self.assertIn(self.responses[0], self.my_survey.responses) def test_store_threee_responses(self): """Test that three repsonses are stored correctly.""" for response in self.responses: self.my_survey.store_responce(response) for response in self.responses: self.assertIn(response, self.my_survey.responses)
from survey import AnonymousSurvey # Define a question, and make a survey question = "What language did you first learn to speak?" my_survey = AnonymousSurvey(question) # Show the question, then store responses to the question my_survey.show_question() print('Enter "q" any time to quit.\n') while True: response = input('Language: ') if response == 'q': break my_survey.store_responce(response) # Show the survey results. print('\nThanks for participating!') my_survey.show_results()