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')