def test_func_get_name(self): comment = Comments(text='text', client_id=1, business_id=1, client_name='client_name', star=1) db.session.add(comment) db.session.commit() self.assertEqual(comment.client_name, Comments.get_name(comment.id))