def test_inter_node_distances_conn(self):
distances = metrics.inter_node_distances(self.g)
desired = {
0: {
1: 1,
2: 1,
3: 2,
4: 3
},
1: {
0: 1,
2: 1,
3: 2,
4: 3
},
2: {
0: 1,
1: 1,
3: 1,
4: 2
},
3: {
0: 2,
1: 2,
2: 1,
4: 1
},
4: {
0: 3,
1: 3,
2: 2,
3: 1
}
}
self.assertEqual(distances, desired)
def test_inter_node_distances_conn(self):
distances = metrics.inter_node_distances(self.g)
desired = {0: {1: 1, 2: 1, 3: 2, 4: 3},
1: {0: 1, 2: 1, 3: 2, 4: 3},
2: {0: 1, 1: 1, 3: 1, 4: 2},
3: {0: 2, 1: 2, 2: 1, 4: 1},
4: {0: 3, 1: 3, 2: 2, 3: 1}}
self.assertEqual(distances, desired)
def test_inter_node_distances_disconn(self):
self.g.remove_edge(2, 3)
# Now all nodes still have at least one edge, but not all nodes are
# reachable from all others.
distances = metrics.inter_node_distances(self.g)
# Distances for all node pairs:
# 0-1: 1 1-2: 1 2-3: Inf 3-4: 1
# 0-2: 1 1-3: Inf 2-4: Inf
# 0-3: Inf 1-4: Inf
# 0-4: Inf
desired = {0: {1: 1, 2: 1, 3: np.inf, 4: np.inf},
1: {0: 1, 2: 1, 3: np.inf, 4: np.inf},
2: {0: 1, 1: 1, 3: np.inf, 4: np.inf},
3: {0: np.inf, 1: np.inf, 2: np.inf, 4: 1},
4: {0: np.inf, 1: np.inf, 2: np.inf, 3: 1}}
self.assertEqual(distances, desired)
def test_inter_node_distances_disconn(self):
self.g.remove_edge(2, 3)
# Now all nodes still have at least one edge, but not all nodes are
# reachable from all others.
distances = metrics.inter_node_distances(self.g)
# Distances for all node pairs:
# 0-1: 1 1-2: 1 2-3: Inf 3-4: 1
# 0-2: 1 1-3: Inf 2-4: Inf
# 0-3: Inf 1-4: Inf
# 0-4: Inf
desired = {
0: {
1: 1,
2: 1,
3: np.inf,
4: np.inf
},
1: {
0: 1,
2: 1,
3: np.inf,
4: np.inf
},
2: {
0: 1,
1: 1,
3: np.inf,
4: np.inf
},
3: {
0: np.inf,
1: np.inf,
2: np.inf,
4: 1
},
4: {
0: np.inf,
1: np.inf,
2: np.inf,
3: 1
}
}
self.assertEqual(distances, desired)
