def test_correctness_on_small_example(self):
r""" Example graph
r
/ \
c1 c2
\ / \
\ / \
c3 c4
"""
O = nx.DiGraph([('c4', 'c2'), ('c3', 'c1'), ('c3', 'c2'), ('c1', 'r'),
('c2', 'r')])
r = set([6])
c1 = set([])
c2 = set([4, 5])
c3 = set([1, 2, 3])
c4 = set([0])
x = {'r': r, 'c1': c1, 'c2': c2, 'c3': c3, 'c4': c4}
b = 'background'
propagate(O, x, b)
self.assertSetEqual(O.node['c3'][b], c3)
self.assertSetEqual(O.node['c2'][b], c4 | c3 | c2)
self.assertSetEqual(O.node['c1'][b], c3 | c1)
self.assertSetEqual(O.node['r'][b], c4 | c3 | c2 | c1 | r)
def test_correctness_on_small_example(self):
r""" Example graph
r
/ \
c1 c2
\ / \
\ / \
c3 c4
"""
O = nx.DiGraph([('c4', 'c2'), ('c3', 'c1'), ('c3', 'c2'),
('c1', 'r'), ('c2', 'r')])
r = set([6])
c1 = set([])
c2 = set([4,5])
c3 = set([1,2,3])
c4 = set([0])
x = { 'r' : r, 'c1' : c1, 'c2' : c2, 'c3' : c3, 'c4' : c4 }
b = 'background'
propagate(O, x, b)
self.assertSetEqual(O.node['c3'][b], c3)
self.assertSetEqual(O.node['c2'][b], c4 | c3 | c2)
self.assertSetEqual(O.node['c1'][b], c3 | c1)
self.assertSetEqual(O.node['r'][b], c4 | c3 | c2 | c1 | r)
def test_if_runs_trough_on_real_data(self):
values = {
k: set(v)
for k, v in self.gene2go.groupby('GO_ID')['GeneID']
}
propagate(self.O, values, 'real')
def test_on_gene2go_head(self):
test = self.gene2go.head(100)
values = {k: set(v) for k, v in test.groupby('GO_ID')['GeneID']}
propagate(self.O, values, 'head')
def test_if_runs_trough_on_real_data(self):
values = {k: set(v) for k,v in self.gene2go.groupby('GO_ID')['GeneID']}
propagate(self.O, values, 'real')
def test_on_gene2go_head(self):
test = self.gene2go.head(100)
values = {k: set(v) for k,v in test.groupby('GO_ID')['GeneID']}
propagate(self.O, values, 'head')
