class TestModels(unittest.TestCase): def setUp(self): self.__test_student = Student(1,"Alin",123) self.__test_assignment = Assignment(2,"lab12","11/11/2034") def test_model_get_set(self): self.assertEqual(self.__test_student.getGroup(),123) self.__test_student.setGroup(134) self.assertEqual(self.__test_student.getGroup() ,134) self.assertEqual(self.__test_assignment.getDescription(),"lab12") self.__test_assignment.setDescription("asd") self.assertEqual(self.__test_assignment.getDescription(),"asd")
def removeAssignedToGroup(self, group): ''' A function that removes a certain assignment that has been assigned to a group, to be used in the undo part. ''' grades = self.getAllGrades() ok = True while ok == True: ok = False for grade in grades: Student = self.__repoStudents.searchByID(grade.getStudentID()) if Student.getGroup() == group: try: self.__repoGrades.remove(grade) ok = True break except RepoError: return