def test_votes(self): question = Question('Are you from India?') question.vote('yes') db.session.add(question) db.session.commit() self.assertEqual(question.number_of_yes_votes, 1) self.assertEqual(question.number_of_no_votes, 0) self.assertEqual(question.number_of_maybe_votes, 0)