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_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']
}
})
