def test_user_follow_unfollow_followers(self): g = NetworkGraph() g.follow('user', 'bernard', 'alex') g.follow('user', 'bernard', 'caroline') g.unfollow('user', 'bernard', 'alex') self.assertEqual(['caroline'], sorted(list(g.get_followers('user', 'bernard'))))
def test_follow_followers(self): g = NetworkGraph() g.set_follow('alex', 'bernard') g.set_follow('caroline', 'bernard') self.assertEqual(['alex', 'caroline'], sorted(list(g.get_followers('bernard'))))