def test_of_three_components(self): """ 1->2 3 4 """ nodes = self.nodes nodes[0].neighbors.append(nodes[1]) self.assertEqual(wcc.weak_connected_components(nodes), [[1, 2], [3], [4]])
def test_of_four_components(self): """ 1 2 3 4 """ nodes = self.nodes self.assertEqual( wcc.weak_connected_components(nodes), [[1], [2], [3], [4]])
def test_of_one_components(self): """ 1->2->3<-4 """ nodes = self.nodes nodes[0].neighbors.append(nodes[1]) nodes[1].neighbors.append(nodes[2]) nodes[3].neighbors.append(nodes[2]) self.assertEqual(wcc.weak_connected_components(nodes), [[1, 2, 3, 4]])