def test_get_num_students_in_program(self):
""" 060A - Valid Get Number of Student in Program """
test_student_1 = Student("Bill", "Smith", "A0100000000", "CIT")
test_student_1.add_course("ACIT2515")
test_student_2 = Student("Ken", "Rodgers", "A0100000001", "CSD")
test_student_2.add_course("COMP1510")
test_student_3 = Student("Sally", "Jones", "A0100000002", "CSD")
test_student_3.add_course("COMP1510")
test_student_4 = Student("Julie", "Wong", "A0100000003", "CSD")
test_student_4.add_course("COMP1510")
test_school = School("Computing and Academic Studies")
test_school.add_student(test_student_1)
test_school.add_student(test_student_2)
test_school.add_student(test_student_3)
test_school.add_student(test_student_4)
self.assertEqual(test_school.get_num_students_in_program("CIT"), 1,
"Must be only 1 CIT student")
self.assertEqual(test_school.get_num_students_in_program("CSD"), 3,
"Must be 3 CSD students")
self.assertEqual(test_school.get_num_students_in_program("SSD"), 0,
"Must be no SSD students")
