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