diff --git a/tests/d2_graph_test.py b/tests/d2_graph_test.py index 21bda74e570148477f7ea6ee4c2d5107eb25c2d0..00d78aeece0cbc6012c9bde992be51068a8021e0 100644 --- a/tests/d2_graph_test.py +++ b/tests/d2_graph_test.py @@ -46,21 +46,35 @@ class TestD2Graph(unittest.TestCase): # Test index awaited_index_size = comb(2*d+1, index_k) + (size - (2*d+1)) * comb(2*d, index_k-1) - # print(d, size, index_k, awaited_index_size) if len(d2.index) != awaited_index_size: dmers = [list(x) for x in d2.index] - dmers = [x for x in dmers if len(x) != len(frozenset(x))] + dmers = [str(x) for x in dmers if len(x) != len(frozenset(x))] - dmers = [str(x) for x in dmers] - # print("\n".join(dmers)) self.assertEquals(awaited_index_size, len(d2.index)) d2_nx = d2.nx_graph - # print(d2_nx.nodes()) - # print(d2_nx.edges()) - # Test connectivity - + # Test connectivity + # Center node names + c1 = d + c2 = d+1 + c3 = d+2 + # Connectivity matrix + awaited_distances = { + c1:{c2:2, c3:4}, + c2:{c1:2, c3:2}, + c3:{c1:4, c2:2} + } + + # distance tests + for idx1, neighbors in d2.distances.items(): + dg1 = d2.node_by_idx[idx1] + + for idx2, dist in neighbors.items(): + dg2 = d2.node_by_idx[idx2] + + awaited_dist = awaited_distances[dg1.center][dg2.center] + self.assertEquals(dist, awaited_dist) if __name__ == "__main__":