def test_all_triads():
"""Tests the all_triplets function."""
G = nx.DiGraph()
G.add_edges_from(['01', '02', '03', '04', '05', '12', '16', '51', '56',
'65'])
expected = [f"{i},{j},{k}" for i in range(7) for j in range(i + 1, 7)
for k in range(j + 1, 7)]
expected = [G.subgraph(x.split(',')) for x in expected]
actual = list(nx.all_triads(G))
assert all(any([nx.is_isomorphic(G1, G2) for G1 in expected])
for G2 in actual)
def test_all_triads():
"""Tests the all_triplets function."""
G = nx.DiGraph()
G.add_edges_from(
["01", "02", "03", "04", "05", "12", "16", "51", "56", "65"])
expected = [
f"{i},{j},{k}" for i in range(7) for j in range(i + 1, 7)
for k in range(j + 1, 7)
]
expected = [G.subgraph(x.split(",")) for x in expected]
actual = list(nx.all_triads(G))
assert all(
any([nx.is_isomorphic(G1, G2) for G1 in expected]) for G2 in actual)
def test_triads_by_type():
"""Tests the all_triplets function."""
G = nx.DiGraph()
G.add_edges_from(["01", "02", "03", "04", "05", "12", "16", "51", "56", "65"])
all_triads = nx.all_triads(G)
expected = defaultdict(list)
for triad in all_triads:
name = nx.triad_type(triad)
expected[name].append(triad)
actual = nx.triads_by_type(G)
assert set(actual.keys()) == set(expected.keys())
for tri_type, actual_Gs in actual.items():
expected_Gs = expected[tri_type]
for a in actual_Gs:
assert any(nx.is_isomorphic(a, e) for e in expected_Gs)
def test_triads_by_type():
"""Tests the all_triplets function."""
G = nx.DiGraph()
G.add_edges_from(['01', '02', '03', '04', '05', '12', '16', '51', '56',
'65'])
all_triads = nx.all_triads(G)
expected = defaultdict(list)
for triad in all_triads:
name = nx.triad_type(triad)
expected[name].append(triad)
actual = nx.triads_by_type(G)
assert set(actual.keys()) == set(expected.keys())
for tri_type, actual_Gs in actual.items():
expected_Gs = expected[tri_type]
for a in actual_Gs:
assert any(nx.is_isomorphic(a, e) for e in expected_Gs)