def test_follow_unfollow(self): user1 = User(name='tester1', email='*****@*****.**', password='******') db.session.add(user1) user2 = User(name='tester2', email='*****@*****.**', password='******') db.session.add(user2) db.session.commit() user1.follow(user2) assert user1.num_following == 1 assert user1.get_following_query().first().id == user2.id assert user2.num_followers == 1 assert user2.get_followers_query().first().id == user1.id user1.unfollow(user2) assert user1.num_following == 0 assert user2.num_followers == 0