def test_invalid_name(): triad_graph('bogus')
def test_triad_graph(): G = triad_graph('030T') assert_equal([tuple(e) for e in ('ab', 'ac', 'cb')], sorted(G.edges()))
def getTriadGraphs(): triad_motifs = [] for i in range(13): if(i == 0): triad_motifs.append(nx.triad_graph("021D")) if(i == 1): triad_motifs.append(nx.triad_graph("021U")) if(i == 2): triad_motifs.append(nx.triad_graph("021C")) if(i == 3): triad_motifs.append(nx.triad_graph("111D")) if(i == 4): triad_motifs.append(nx.triad_graph("111U")) if(i == 5): triad_motifs.append(nx.triad_graph("030T")) if(i == 6): triad_motifs.append(nx.triad_graph("030C")) if(i == 7): triad_motifs.append(nx.triad_graph("201")) if(i == 8): triad_motifs.append(nx.triad_graph("120D")) if(i == 9): triad_motifs.append(nx.triad_graph("120U")) if(i == 10): triad_motifs.append(nx.triad_graph("120C")) if(i == 11): triad_motifs.append(nx.triad_graph("210")) if(i == 12): triad_motifs.append(nx.triad_graph("300")) return triad_motifs
def test_triad_graph(): G = triad_graph("030T") assert [tuple(e) for e in ("ab", "ac", "cb")] == sorted(G.edges())
def test_invalid_name(): with pytest.raises(ValueError): triad_graph("bogus")
compdata['result'] = compdata['p'] < compdata['comp'] compdata['count'] = empirical_triads print(compdata) plt.rcParams['font.size'] = 12 fig, axs = plt.subplots(4, 4, figsize=(10, 10)) pos = {'a': (1, 1), 'b': (6, 1), 'c': (3.5, 5.333)} # scaling... newpos = {a: (b[0] / 10, b[1] / 10) for a, b in pos.items()} # empirical counts are in empirical_triads, indexed by triad name # p-values are in ps, indexed by triad name triads = list(empirical_triads.index) for ax, triad in zip(axs.flatten(), triads): g = nx.triad_graph(triad) nx.draw(g, pos=newpos, with_labels=False, ax=ax, node_size=50, width=.5, node_color='k') ax.set_title(triad) ax.text(newpos['a'][0], newpos['a'][1] - 0.05, r'$n$ = ' + f'{empirical_triads.loc[triad]}', horizontalalignment='left', verticalalignment='top') ax.text(newpos['b'][0],