def test_unpack_treeset(self): struct = OOBTree.OOTreeSet() struct.insert('foo') struct.insert('bar') g = NetworkGraph() flat = g.unpack(struct) self.assertEqual(flat, ['bar', 'foo'])
def test_get_tags(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('content', 'doc1', 'caroline', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') tags = g.unpack(g.get_tags('content', 'doc1', 'caroline')) self.assertEqual(tags, ['leadership', 'negotiations'])
def test_unpack_btree_treeset(self): struct = OOBTree.OOBTree() struct['foo'] = OOBTree.OOTreeSet() struct['foo'].insert('baz') struct['foo'].insert('bar') g = NetworkGraph() flat = g.unpack(struct) self.assertEqual(flat, {'foo': ['bar', 'baz']})
def test_get_taggers_notag(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') # an extra tag here g.tag('user', 'bernard', 'caroline', 'negotiations') taggers = g.unpack(g.get_taggers('user', 'bernard')) self.assertEqual(taggers, {'change management': ['alex'], 'leadership': ['alex'], 'negotiations': ['caroline']})
def test_get_taggers_notag(self): g = NetworkGraph() g.tag('user', 'bernard', 'alex', 'leadership', 'change management') g.tag('user', 'caroline', 'alex', 'leadership', 'negotiations') g.tag('content', 'doc1', 'alex', 'negotiations') # an extra tag here g.tag('user', 'bernard', 'caroline', 'negotiations') taggers = g.unpack(g.get_taggers('user', 'bernard')) self.assertEqual( taggers, { 'change management': ['alex'], 'leadership': ['alex'], 'negotiations': ['caroline'] })