def test_panther_similarity_weighted(self):
numpy.random.seed(42)
G = nx.Graph()
G.add_edge("v1", "v2", weight=5)
G.add_edge("v1", "v3", weight=1)
G.add_edge("v1", "v4", weight=2)
G.add_edge("v2", "v3", weight=0.1)
G.add_edge("v3", "v5", weight=1)
expected = {"v3": 0.75, "v4": 0.5, "v2": 0.5, "v5": 0.25}
sim = nx.panther_similarity(G, "v1", path_length=2)
assert sim == expected
def test_panther_similarity_unweighted(self):
numpy.random.seed(42)
G = nx.Graph()
G.add_edge(0, 1)
G.add_edge(0, 2)
G.add_edge(0, 3)
G.add_edge(1, 2)
G.add_edge(2, 4)
expected = {3: 0.5, 2: 0.5, 1: 0.5, 4: 0.125}
sim = nx.panther_similarity(G, 0, path_length=2)
assert sim == expected
