5
5
6
6
7
7
class TestGraph (unittest .TestCase ):
8
-
9
8
def assert_out_mapping (self , g , expected_out_mapping ):
10
9
self .assertDictEqual (g .out_mapping , expected_out_mapping )
11
10
for node , out_dict in expected_out_mapping .items ():
@@ -97,7 +96,6 @@ def test_add_loops_with_existing_loop_and_using_strings(self):
97
96
98
97
99
98
class TestCycle (unittest .TestCase ):
100
-
101
99
def test_length_1_directed (self ):
102
100
g = graph .cycle (1 , directed = True )
103
101
self .assertEqual (g .vertices , [0 ])
@@ -124,7 +122,7 @@ def test_length_3_directed(self):
124
122
g = graph .cycle (3 , directed = True )
125
123
self .assertEqual (g .vertices , [0 , 1 , 2 ])
126
124
self .assertEqual (g .edges , [(0 , 1 ), (1 , 2 ), (2 , 0 )])
127
-
125
+
128
126
def test_length_3_undirected (self ):
129
127
g = graph .cycle (3 , directed = False )
130
128
edges = [(0 , 1 ), (1 , 0 ), (1 , 2 ), (2 , 1 ), (2 , 0 ), (0 , 2 )]
@@ -156,7 +154,6 @@ def test_length_4_undirected(self):
156
154
157
155
158
156
class TestComplete (unittest .TestCase ):
159
-
160
157
def test_size_2 (self ):
161
158
g = graph .complete_graph (2 , loops = False )
162
159
self .assertEqual (g .vertices , [0 , 1 ])
@@ -209,7 +206,7 @@ def test_size_2_with_loops(self):
209
206
self .assertEqual (g .vertices , [0 , 1 ])
210
207
self .assertEqual (g .edges , [(0 , 1 ), (1 , 0 ), (0 , 0 ), (1 , 1 )])
211
208
self .assertEqual (g .directed , False )
212
-
209
+
213
210
def test_size_3_with_loops (self ):
214
211
g = graph .complete_graph (3 , loops = True )
215
212
self .assertEqual (g .vertices , [0 , 1 , 2 ])
0 commit comments