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