def test_all_triplets(): """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 = [set(x.split(',')) for x in expected] actual = list(set(x) for x in nx.all_triplets(G)) assert all([any([s1 == s2 for s1 in expected]) for s2 in actual])
def test_all_triplets(): """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 = [set(x.split(",")) for x in expected] actual = list(set(x) for x in nx.all_triplets(G)) assert all([any([s1 == s2 for s1 in expected]) for s2 in actual])