Python: grade_school

python/grade-school/grade_school_test.py

@@ -0,0 +1,64 @@
+import unittest
+
+from grade_school import School
+
+# Tests adapted from `problem-specifications//canonical-data.json`
+
+
+class GradeSchoolTest(unittest.TestCase):
+    def test_adding_a_student_adds_them_to_the_sorted_roster(self):
+        school = School()
+        school.add_student(name="Aimee", grade=2)
+        expected = ["Aimee"]
+        self.assertEqual(school.roster(), expected)
+
+    def test_adding_more_student_adds_them_to_the_sorted_roster(self):
+        school = School()
+        school.add_student(name="Blair", grade=2)
+        school.add_student(name="James", grade=2)
+        school.add_student(name="Paul", grade=2)
+        expected = ["Blair", "James", "Paul"]
+        self.assertEqual(school.roster(), expected)
+
+    def test_adding_students_to_different_grades_adds_them_to_the_same_sorted_roster(
+        self
+    ):
+        school = School()
+        school.add_student(name="Chelsea", grade=3)
+        school.add_student(name="Logan", grade=7)
+        expected = ["Chelsea", "Logan"]
+        self.assertEqual(school.roster(), expected)
+
+    def test_roster_returns_an_empty_list_if_there_are_no_students_enrolled(self):
+        school = School()
+        expected = []
+        self.assertEqual(school.roster(), expected)
+
+    def test_student_names_with_grades_are_displayed_in_the_same_sorted_roster(self):
+        school = School()
+        school.add_student(name="Peter", grade=2)
+        school.add_student(name="Anna", grade=1)
+        school.add_student(name="Barb", grade=1)
+        school.add_student(name="Zoe", grade=2)
+        school.add_student(name="Alex", grade=2)
+        school.add_student(name="Jim", grade=3)
+        school.add_student(name="Charlie", grade=1)
+        expected = ["Anna", "Barb", "Charlie", "Alex", "Peter", "Zoe", "Jim"]
+        self.assertEqual(school.roster(), expected)
+
+    def test_grade_returns_the_students_in_that_grade_in_alphabetical_order(self):
+        school = School()
+        school.add_student(name="Franklin", grade=5)
+        school.add_student(name="Bradley", grade=5)
+        school.add_student(name="Jeff", grade=1)
+        expected = ["Bradley", "Franklin"]
+        self.assertEqual(school.grade(5), expected)
+
+    def test_grade_returns_an_empty_list_if_there_are_no_students_in_that_grade(self):
+        school = School()
+        expected = []
+        self.assertEqual(school.grade(1), expected)
+
+
+if __name__ == "__main__":
+    unittest.main()