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
