def test_follow_unfollow_followers(self): g = NetworkGraph() g.set_follow('alex', 'bernard') g.set_follow('caroline', 'bernard') g.set_unfollow('alex', 'bernard') self.assertEqual(['caroline'], sorted(list(g.get_followers('bernard'))))
def test_follow_following(self): g = NetworkGraph() g.set_follow('alex', 'bernard') g.set_follow('alex', 'caroline') g.set_follow('alex', 'dick') self.assertEqual(['bernard', 'caroline', 'dick'], sorted(list(g.get_following('alex'))))
def test_follow_unfollow_following(self): g = NetworkGraph() g.set_follow('alex', 'bernard') g.set_follow('alex', 'caroline') g.set_unfollow('alex', 'bernard') self.assertEqual(['caroline'], list(g.get_following('alex')))
def test_follow(self): g = NetworkGraph() g.set_follow('alex', 'bernard') self.assertEqual(['bernard'], list(g.get_following('alex')))