def test_should_be_able_to_create_directed_network_from_adj_matrix():
adj_matrix = np.array([[0, 2, 0, 0, 1, 0], [1, 0, 1, 1, 0, 0],
[0, 1, 4, 1, 1, 1], [0, 1, 1, 0, 0, 0],
[1, 0, 1, 0, 0, 0], [0, 0, 1, 0, 0, 2]])
dn_from_adj = n.DirectedNetwork(adj_matrix)
dn = n.DirectedNetwork()
for node in range(1, 7):
dn.add_node(0, str(node))
dn.add_edge("1", "2")
dn.add_edge("1", "2")
dn.add_edge("1", "5")
dn.add_edge("2", "1")
dn.add_edge("2", "3")
dn.add_edge("2", "4")
dn.add_edge("3", "2")
dn.add_edge("3", "3")
dn.add_edge("3", "3")
dn.add_edge("3", "4")
dn.add_edge("3", "5")
dn.add_edge("3", "6")
dn.add_edge("4", "2")
dn.add_edge("4", "3")
dn.add_edge("5", "1")
dn.add_edge("5", "3")
dn.add_edge("6", "3")
dn.add_edge("6", "6")
assert dn_from_adj == dn
assert (adj_matrix == dn.adjacency_matrix).all()
def test_create_nodes_and_edge_should_be_equal_to_adding_nodes_and_edge():
dn_one = n.DirectedNetwork()
dn_one.add_node(1)
dn_one.add_node(2)
dn_one.add_edge(1, 2)
dn_two = n.DirectedNetwork()
dn_two.create_nodes_and_edge(1, 2)
assert dn_one == dn_two
def test_add_edge_and_remove_edge_should_leave_network_in_unchanged_state():
dn = n.DirectedNetwork()
dn.add_node(1)
dn.add_node(2)
copied_dn = copy.deepcopy(dn)
dn.add_edge(1, 2)
dn.remove_edge(1, 2)
assert dn == copied_dn
def test_network_should_remember_order_of_insertion_even_when_element_removed(
):
dn = n.DirectedNetwork()
dn.add_node(3)
dn.add_node(2)
dn.add_node(5)
dn.remove_node(2)
assert list(dn.nodes.keys()) == ['3', '5']
def test_adding_and_removing_node_should_leave_network_in_unchanged_state():
dn = n.DirectedNetwork()
dn.add_node(1)
copied_dn = copy.deepcopy(dn)
dn.add_node(2)
removed = dn.remove_node(2)
assert removed is True
assert dn == copied_dn
def test_create_nodes_and_edge_should_be_idempotent():
dn = n.DirectedNetwork()
dn.create_nodes_and_edge(1, 2)
copied_dn = copy.deepcopy(dn)
copied_dn.create_nodes_and_edge(1, 2)
assert dn == copied_dn
def test_add_edge_should_throw_an_error_if_to_node_does_not_exist():
dn = n.DirectedNetwork()
dn.add_node(2)
with pytest.raises(ValueError):
dn.add_edge(1, 2)
def test_add_edge_should_connect_two_nodes():
dn = n.DirectedNetwork()
dn.add_node(1)
dn.add_node(2)
dn.add_edge(1, 2)
assert dn.edges['1'] == list('2')
def test_removing_non_existing_node_should_throw_an_error():
dn = n.DirectedNetwork()
assert dn.remove_node(2) is False
def test_directed_add_node_should_add_node():
dn = n.DirectedNetwork()
dn.add_node(1)
assert len(dn.nodes) == 1
assert nodes_equal(dn.nodes['1'], tn.Node(1, name="1"))
def test_directed_add_node_should_throw_if_node_already_exist():
dn = n.DirectedNetwork()
dn.add_node(1)
with pytest.raises(ValueError):
dn.add_node(1)
def test_should_create_directed_network_from_single_entry():
dn = n.DirectedNetwork(np.array([0]))
assert len(dn.nodes) == 1
assert nodes_equal(dn.nodes['1'], tn.Node(0, name="1"))
assert len(dn.edges) == 0
def test_directed_network_should_throw_on_non_even_entry_in_diagonal():
with pytest.raises(ValueError):
n.DirectedNetwork(np.array([[1, 0], [0, 0]]))
def test_directed_network_should_throw_on_non_square_adj_matrix():
with pytest.raises(ValueError):
n.DirectedNetwork(np.array([[1, 2, 3], [3, 2, 1]]))
def test_add_node_unique_should_add_new_node_every_time():
dn = n.DirectedNetwork()
dn.add_node_unique(1)
dn.add_node_unique(1)
dn.add_node_unique(1)
assert len(dn.nodes) == 3
def test_network_should_remember_order_of_insertion():
dn = n.DirectedNetwork()
dn.add_node(3)
dn.add_node(2)
dn.add_node(5)
assert list(dn.nodes.keys()) == ['3', '2', '5']
import network_science.ns.network as n
import numpy as np
import copy
adj_matrix = np.array([[0, 2, 0, 0, 1, 0], [1, 0, 1, 1, 0, 0],
[0, 1, 4, 1, 1, 1], [0, 1, 1, 0, 0, 0],
[1, 0, 1, 0, 0, 0], [0, 0, 1, 0, 0, 2]])
dn_from_adj = n.DirectedNetwork(adj_matrix)
dn = n.DirectedNetwork()
for node in range(1, 7):
dn.add_node(0, str(node))
dn.add_edge("1", "2")
dn.add_edge("1", "2")
dn.add_edge("1", "5")
dn.add_edge("2", "1")
dn.add_edge("2", "3")
dn.add_edge("2", "4")
dn.add_edge("3", "2")
dn.add_edge("3", "3")
dn.add_edge("3", "3")
dn.add_edge("3", "4")
dn.add_edge("3", "5")
dn.add_edge("3", "6")
dn.add_edge("4", "2")
dn.add_edge("4", "3")
dn.add_edge("5", "1")
