class TestSurvey(unittest.TestCase): def setUp(self): question = 'Why?' self.survey = Survey(question) def test_insert_response_single(self): self.survey.insert_response('English') self.assertIn('English', self.survey.responses) def test_insert_response_multi(self): response_sample = ['English', 'Chinese', 'Japanese'] for item in response_sample: self.survey.insert_response(item) for response in response_sample: self.assertIn(response, self.survey.responses)
# from name_function import get_full_name # str = get_full_name('zhang', 'bin') # print(str) #--- from survey import Survey question = '\nWhat is your favorite color?' survey = Survey(question) while True: result = input(f"{question} Enter 'q' to quit\n") if result == 'q': break else: survey.insert_response(result) survey.report()