def test_add_remove_change_rating(self): new_rating = Rating(rating=2.5, user_id=11111, drink_id=1) new_rating.id = 10000 db.session.add(new_rating) rating = Rating.query.get(10000) self.assertEqual(rating.id, 10000) self.assertEqual(rating.rating, 2.5) self.assertEqual(rating.user_id, 11111) self.assertEqual(rating.drink_id, 1) rating.rating = 1.0 db.session.commit() self.assertEqual(rating.rating, 1.0) db.session.delete(rating) db.session.commit() check_rating = Rating.query.get(10000) self.assertEqual(check_rating, None)