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]
)
