def extract_hier_all(gosubdag, out, root_term, go2geneids, prize_dictionary):
"""write_hier.py: Prints the entire mini GO hierarchy, with counts of children."""
out.write('\nTEST EXTRACTION: Print all hierarchies:\n')
objwr = WrHierGO(gosubdag,
go2geneids=go2geneids,
prize_dictionary=prize_dictionary,
G=nx.DiGraph())
obj = objwr.ext_hier_down(root_term, out)
seeds = prize_dictionary.keys()
layer_i_nodes = linear_threshold(obj.G, seeds, steps=INFLUENCE_N_STEPS)
return (obj.vertices, obj.edges, obj.G, layer_i_nodes)
def extract_hier_all(gosubdag, out, root_term, go2geneids):
"""write_hier.py: Prints the entire mini GO hierarchy, with counts of children."""
out.write('\nTEST EXTRACTION: Print all hierarchies:\n')
objwr = WrHierGO(gosubdag, go2geneids=go2geneids)
obj = objwr.ext_hier_down(root_term, out)
return (obj.vertices, obj.edges)