def test_node_labels_subgraph(): adj_matrix = [[0, 1, 0], [0, 0, 1], [1, 0, 0]] node_labels = np.array(['a', 'b', 'c']) g = DiGraph(adj_matrix, node_labels=node_labels) nodes = [1, 2] assert_array_equal(g.subgraph(nodes).node_labels, node_labels[nodes])
def test_subgraph_weighted(): adj_matrix = np.arange(3**2).reshape(3, 3) g = DiGraph(adj_matrix, weighted=True) nodes = [0, 1] subgraph_adj_matrix = [[0, 1], [3, 4]] assert_array_equal( g.subgraph(nodes).csgraph.toarray(), subgraph_adj_matrix)
def test_subgraph(): adj_matrix = [[0, 1, 0], [0, 0, 1], [1, 0, 0]] g = DiGraph(adj_matrix) nodes = [1, 2] subgraph_adj_matrix = [[False, True], [False, False]] assert_array_equal( g.subgraph(nodes).csgraph.toarray(), subgraph_adj_matrix)
def test_subgraph_weighted(): adj_matrix = np.arange(3**2).reshape(3, 3) g = DiGraph(adj_matrix, weighted=True) nodes = [0, 1] subgraph_adj_matrix = [[0, 1], [3, 4]] assert_array_equal( g.subgraph(nodes).csgraph.toarray(), subgraph_adj_matrix )
def test_subgraph(): adj_matrix = [[0, 1, 0], [0, 0, 1], [1, 0, 0]] g = DiGraph(adj_matrix) nodes = [1, 2] subgraph_adj_matrix = [[False, True], [False, False]] assert_array_equal( g.subgraph(nodes).csgraph.toarray(), subgraph_adj_matrix )
def test_node_labels_subgraph(): adj_matrix = [[0, 1, 0], [0, 0, 1], [1, 0, 0]] node_labels = np.array(['a', 'b', 'c']) g = DiGraph(adj_matrix, node_labels=node_labels) nodes = [1, 2] assert_array_equal( g.subgraph(nodes).node_labels, node_labels[nodes] )