@@ -94,7 +94,8 @@ def test_out_dict(self):
94
94
2 : {1 : None , 3 : None },
95
95
3 : {2 : None }
96
96
}
97
- self .assertEqual (g .out_mapping , expected_out_mapping )
97
+ for key in expected_out_mapping :
98
+ self .assertEqual (g .out_dict (key ), expected_out_mapping [key ])
98
99
99
100
# Undirected graph with vertices and weighted edges
100
101
g = graph .Graph (edges = [[1 , 2 , 10 ], [2 , 3 , 5 ]])
@@ -103,15 +104,17 @@ def test_out_dict(self):
103
104
2 : {1 : 10 , 3 : 5 },
104
105
3 : {2 : 5 }
105
106
}
106
- self .assertEqual (g .out_mapping , expected_out_mapping )
107
+ for key in expected_out_mapping :
108
+ self .assertEqual (g .out_dict (key ), expected_out_mapping [key ])
107
109
108
110
# Directed graph with vertices and weighted edges
109
111
g = graph .Graph (edges = [[1 , 2 , 10 ], [2 , 3 , 5 ]], directed = True )
110
112
expected_out_mapping = {
111
113
1 : {2 : 10 },
112
114
2 : {3 : 5 },
113
115
}
114
- self .assertEqual (g .out_mapping , expected_out_mapping )
116
+ for key in expected_out_mapping :
117
+ self .assertEqual (g .out_dict (key ), expected_out_mapping [key ])
115
118
116
119
def test_in_dict (self ):
117
120
# Undirected graph with vertices and unweighted edges
@@ -121,7 +124,8 @@ def test_in_dict(self):
121
124
2 : {1 : None , 3 : None },
122
125
3 : {2 : None }
123
126
}
124
- self .assertEqual (g .in_mapping , expected_in_mapping )
127
+ for key in expected_in_mapping :
128
+ self .assertEqual (g .in_dict (key ), expected_in_mapping [key ])
125
129
126
130
# Undirected graph with vertices and weighted edges
127
131
g = graph .Graph (edges = [[1 , 2 , 10 ], [2 , 3 , 5 ]])
@@ -130,15 +134,17 @@ def test_in_dict(self):
130
134
2 : {1 : 10 , 3 : 5 },
131
135
3 : {2 : 5 }
132
136
}
133
- self .assertEqual (g .in_mapping , expected_in_mapping )
137
+ for key in expected_in_mapping :
138
+ self .assertEqual (g .in_dict (key ), expected_in_mapping [key ])
134
139
135
140
# Directed graph with vertices and weighted edges
136
141
g = graph .Graph (edges = [[1 , 2 , 10 ], [2 , 3 , 5 ]], directed = True )
137
142
expected_in_mapping = {
138
143
2 : {1 : 10 },
139
144
3 : {2 : 5 }
140
145
}
141
- self .assertEqual (g .in_mapping , expected_in_mapping )
146
+ for key in expected_in_mapping :
147
+ self .assertEqual (g .in_dict (key ), expected_in_mapping [key ])
142
148
143
149
def test_repr (self ):
144
150
# Undirected graph with no vertices
0 commit comments