def test_get_non_existent_course(self):
""" 050C - Invalid Get Course Non-Existent """
test_course_1 = Course("ACIT2515", "123456", "CIT")
test_course_1.add_student("A010000056")
test_course_2 = Course("COMP1510", "456321", "CST")
test_course_2.add_student("A010000056")
test_course_2.add_student("A010450012")
test_school = School("Computing and Academic Studies")
test_school.add_course(test_course_1)
test_school.add_course(test_course_2)
self.assertIsNone(test_school.get_course("ACIT1234"), "No course should exists for ACIT1234")
def test_get_course(self):
""" 050A - Valid Get Course """
test_course_1 = Course("ACIT2515", "123456", "CIT")
test_course_1.add_student("A010000056")
test_course_2 = Course("COMP1510", "456321", "CST")
test_course_2.add_student("A010000056")
test_course_2.add_student("A010450012")
test_school = School("Computing and Academic Studies")
test_school.add_course(test_course_1)
test_school.add_course(test_course_2)
retrieved_course = test_school.get_course("ACIT2515")
self.assertEqual(retrieved_course.get_course_id(), "ACIT2515", "Course must have course ID ACIT2515")
self.assertEqual(retrieved_course.get_crn(), "123456", "Course must have CRN 123456")
self.assertEqual(retrieved_course.get_program(), "CIT", "Course must be in CIT program")
