def test_score(self): staff = Staff() Comment1 = Comment(score=3, staff=staff) Comment2 = Comment(score=4, staff=staff) staff.update_score() self.assertEqual(staff.score, 3.5)