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],
