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_string_args(self): """BTree keys MUST be of same type. Check that the implementation enforces this.""" g = NetworkGraph() self.assertRaises(AssertionError, g.set_follow, 1, '2') self.assertRaises(AssertionError, g.set_follow, '1', 2) self.assertRaises(AssertionError, g.set_unfollow, 1, '2') self.assertRaises(AssertionError, g.set_unfollow, '1', 2) self.assertRaises(AssertionError, g.get_following, 2) self.assertRaises(AssertionError, g.get_followers, 2)
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')))