def test_remove_student_not_in_class_student_error(self):
test_class = ClassList(2)
test_class.add_student('Test Student')
test_class.add_student('Joe')
with self.assertRaises(StudentError):
test_class.remove_student('Emily')
def test_remove_student_from_empty_list_student_error(self):
test_class = ClassList(2)
with self.assertRaises(StudentError):
test_class.remove_student('Emily')
def test_add_remove_student_ensure_removed(self):
test_class = ClassList(2)
test_class.add_student('Test Student')
test_class.remove_student('Test Student')
self.assertNotIn('Test Student', test_class.class_list)