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