Chapter End

Author: Hridoy-31

11. Testing Your Code/language_survey.py

@@ -0,0 +1,16 @@
+from survey import AnonymousSurvey
+
+question = "What language did you first learn to speak ? "
+my_survey = AnonymousSurvey(question)
+
+my_survey.show_question()
+print("Enter 'q' to quit it.")
+
+while True:
+    response = input("Language: ")
+    if response == 'q':
+        break
+    my_survey.store_response(response)
+
+print("Thank you for participating in the survey ")
+my_survey.show_results()

11. Testing Your Code/survey.py

@@ -4,11 +4,18 @@ class AnonymousSurvey:
     def __init__(self, question):
         """ Store a qiestion and prepare to store the response """
         self.question = question
-        self.response = []
-             
+        self.responses = []
+
     def show_question(self):
         """ to show the question of the survey """
         print(self.question)
 
+    def store_response(self, new_response):
+        """ storing the responses from the input """
+        self.responses.append(new_response)
 
-    
+    def show_results(self):
+        """ Showing the result """
+        print("Survey results: ")
+        for response in self.responses:
+            print(f"- {response}")

11. Testing Your Code/test_survey.py

@@ -0,0 +1,27 @@
+import unittest
+from survey import AnonymousSurvey
+
+
+class TestAnonymousSurvey(unittest.TestCase):
+    """Tests for the class Anonymous Survey"""
+
+    def setUp(self):
+        """ set up survey and responses for further uses """
+        question = "What language did you first learn to speak ?"
+        self.my_survey = AnonymousSurvey(question)
+        self.responses = ['English', 'Spanish', 'Mandarin']
+
+    def test_store_single_responses(self):
+        """ Single response testing """
+        self.my_survey.store_response(self.responses[0])
+        self.assertIn(self.responses[0], self.my_survey.responses)
+
+    def test_store_triple_responses(self):
+        """ Triple response testing """
+        for response in self.responses:
+            self.my_survey.store_response(response)
+            self.assertIn(response, self.my_survey.responses)
+
+
+if __name__ == "__main__":
+    unittest.main()