def test_can_remove_exercise(self):
assignment = Assignment()
exercise = MagicMock()
assignment.add_exercise(exercise)
assignment.remove_exercise(exercise)
number_of_exercises = len(assignment.exercises)
assert number_of_exercises == 0
def test_can_remove_exercise(self):
assignment = Assignment()
exercise = MagicMock()
assignment.add_exercise(exercise)
assignment.remove_exercise(exercise)
number_of_exercises = len(assignment.exercises)
assert number_of_exercises == 0
def test_score_is_updated_when_exercises_are_removed(self):
assignment = Assignment()
exercise = MagicMock()
exercise.score = 30
exercise_to_be_removed = MagicMock()
exercise_to_be_removed.score = 10
assignment.add_exercise(exercise)
assignment.add_exercise(exercise_to_be_removed)
assignment.remove_exercise(exercise_to_be_removed)
assert assignment.score == 30
def test_score_is_updated_when_exercises_are_removed(self):
assignment = Assignment()
exercise = MagicMock()
exercise.score = 30
exercise_to_be_removed = MagicMock()
exercise_to_be_removed.score = 10
assignment.add_exercise(exercise)
assignment.add_exercise(exercise_to_be_removed)
assignment.remove_exercise(exercise_to_be_removed)
assert assignment.score == 30
