def test_get_tagged_all(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tagged = sorted(list(g.get_tagged('user', 'alex', 'leadership'))) self.assertEqual(tagged, ['bernard', 'caroline']) tagged = sorted(list(g.get_tagged('content', 'alex', 'negotiations'))) self.assertEqual(tagged, ['doc1'])
def test_get_tagged_nouserid(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tagged = g.get_tagged('user', None, 'leadership') self.assertEqual(tagged, {'bernard': ['alex'], 'caroline': ['alex']})
def test_get_tagged_nouserid(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tagged = g.get_tagged('user', None, 'leadership') self.assertEqual( tagged, {'bernard': ['alex'], 'caroline': ['alex']} )
def test_get_tagged_noparams(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tagged = g.get_tagged() self.assertEqual( tagged, {'alex': {'change management': {'content': [], 'user': ['bernard']}, 'leadership': {'content': [], 'user': ['bernard', 'caroline']}, 'negotiations': {'content': ['doc1'], 'user': ['caroline']}}} )
def test_get_tagged_only_user_id(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tagged = g.get_tagged(None, 'alex', None) self.assertEqual( tagged, { 'change management': { 'content': [], 'user': ['bernard'] }, 'leadership': { 'content': [], 'user': ['bernard', 'caroline'] }, 'negotiations': { 'content': ['doc1'], 'user': ['caroline'] } })