def test_sub_networks():
"""Test to remove a network"""
net_1 = Network()
net_2 = Network()
net_1.add_edge('a', 'b', uid='a-b')
net_2.add_edge('c', 'd', uid='c-d')
net_1 += net_2
net_2.add_edge('d', 'e', uid='d-e')
net_3 = net_1 - net_2
assert net_3.number_of_nodes() == 2
assert net_3.number_of_edges() == 1
assert 'a' and 'b' in net_3.nodes
assert 'a-b' in net_3.edges
assert net_1.number_of_nodes() == 4
assert net_1.number_of_edges() == 2
assert net_2.number_of_nodes() == 3
assert net_2.number_of_edges() == 2
net_4 = Network()
net_4.add_edge('x', 'y', uid='x-y')
net_5 = net_3 - net_4
assert net_5.number_of_nodes() == 2
assert net_5.number_of_edges() == 1
assert 'a' and 'b' in net_5.nodes
assert 'a-b' in net_5.edges
def test_add_node():
"""Test the node assignment."""
# add string and create new node in the network
net = Network()
net.add_node('v', color='red')
assert len(net.nodes) == 1
assert 'v' in net.nodes.uids
assert net.nodes['v']['color'] == 'red'
assert net.nodes.index['v'] == 0
assert net.number_of_nodes() == 1
assert isinstance(net.nodes['v'], Node)
assert net.nodes['v'].uid == 'v'
w = Node('w', color='green')
net.add_node(w)
assert net.number_of_nodes() == 2
assert isinstance(net.nodes['w'], Node)
assert net.nodes['w'].uid == 'w'
assert net.nodes['w']['color'] == 'green'
v = Node('v', color='blue')
with pytest.raises(Exception):
net.add_node(v)
def test_add_edges():
"""Test assigning edges form a list."""
net = Network()
ab = Edge(Node('a'), Node('b'))
net.add_edges(ab, ('b', 'c'))
assert net.number_of_edges() == 2
net = Network()
edges = [("A", "B"), ("B", "C")]
net.add_edges(edges)
assert net.number_of_edges() == 2
assert net.number_of_nodes() == 3
net = Network()
edges = [("a", "b"),
("b", "c"),
("c", "d"),
("c", "e")]
edges = [tuple(Node(x) for x in e) for e in edges]
# with pytest.raises(Exception):
# net.add_edges(edges)
net = Network()
net.add_edges(('a', 'b'), ('b', 'c'))
def test_isub_networks():
"""Test to remove a network with isub"""
net_1 = Network()
net_2 = Network()
net_1.add_edge('a', 'b', uid='a-b')
net_2.add_edge('c', 'd', uid='c-d')
net_1 += net_2
net_2.add_edge('d', 'e', uid='d-e')
net_1 -= net_2
assert net_1.number_of_nodes() == 2
assert net_1.number_of_edges() == 1
assert 'a' and 'b' in net_1.nodes
assert 'a-b' in net_1.edges
assert net_2.number_of_nodes() == 3
assert net_2.number_of_edges() == 2
def test_add_nodes():
"""Test assigning notes form a list."""
net = Network()
u = Node('u', color='blue')
net.add_nodes(u, 'v', 'w', color='green')
assert net.number_of_nodes() == 3
assert net.nodes['u']['color'] == 'green'
assert net.nodes['v']['color'] == 'green'
assert net.nodes['w']['color'] == 'green'
def test_sql_write_network():
"""Write network to sql database."""
net = Network()
net.add_edges(('a', 'b'), ('a', 'c'))
pp.io.sql.write(net,
filename='network.db',
table='test',
if_exists='replace')
net = pp.io.sql.read_network(db_file='network.db', table='test')
assert isinstance(net, Network)
assert net.number_of_nodes() == 3
assert net.number_of_edges() == 2
def get_metric(network: pp.Network, metric: str):
switcher = {
'num_vertices':
network.number_of_nodes(),
'num_edges':
network.number_of_edges(),
'is_directed':
network.directed,
'average_degree':
pp.statistics.mean_degree(network),
'degree_assortativity':
pp.statistics.degree_assortativity(network),
'global_clustering_coefficient':
pp.statistics.avg_clustering_coefficient(network),
'diameter':
pp.algorithms.diameter(network),
'edge_reciprocity':
pp.statistics.edge_reciprocity(network),
'largest_component_fraction':
pp.algorithms.largest_component_size(network) /
network.number_of_nodes()
}
return switcher.get(metric, None)
def test_iadd_networks():
"""Test to add networks together"""
net_1 = Network()
net_1.add_edges(('a', 'b'), ('b', 'c'))
net_2 = Network()
net_2.add_edges(('x', 'y'), ('y', 'z'))
net_1 += net_2
assert net_1.number_of_nodes() == 6
assert net_1.number_of_edges() == 4
assert net_2.number_of_nodes() == 3
assert net_2.number_of_edges() == 2
# test same node objects
a = Node('a')
b = Node('b')
c = Node('c')
net_1 = Network()
net_2 = Network()
net_1.add_edge(a, b)
net_2.add_edge(b, c)
net_1 += net_2
assert net_1.number_of_nodes() == 3
assert net_1.number_of_edges() == 2
assert net_2.number_of_nodes() == 2
assert net_2.number_of_edges() == 1
# nodes with same uids but different objects
net_1 = Network()
net_2 = Network()
net_1.add_edge(a, b)
net_2.add_edge('b', c)
with pytest.raises(Exception):
net_1 += net_2
# test same edge objects
a = Node('a')
b = Node('b')
c = Node('c')
net_1 = Network()
net_2 = Network()
net_1.add_edge(a, b, uid='e1')
net_2.add_edge(a, b, uid='e2')
with pytest.raises(Exception):
net_1 += net_2
# assert net_1.number_of_edges() == 2
# assert net_1.number_of_nodes() == 2
# assert 'e1' in net_1.edges and 'e2' in net_1.edges
# edges with same uids but different objects
net_1 = Network()
net_2 = Network()
net_1.add_edge(a, b, uid='e1')
net_2.add_edge(a, b, uid='e1')
with pytest.raises(Exception):
net_1 += net_2
# add multiple networks
net_1 = Network()
net_2 = Network()
net_3 = Network()
net_1.add_edge('a', 'b')
net_2.add_edge('c', 'd')
net_3.add_edge('e', 'f')
net_1 += net_2 + net_3
assert net_1.number_of_edges() == 3
assert net_1.number_of_nodes() == 6
def test_add_edge():
"""Test the edge assignment."""
a = Node('a')
b = Node('b')
c = Node('c')
# add edges with no uids
e = Edge(a, b)
f = Edge(b, c)
g = Edge(a, b)
net = Network()
net.add_edge(e)
net.add_edge(f)
with pytest.raises(Exception):
net.add_edge(g)
assert len(net.edges) == 2
assert len(net.nodes) == 3
with pytest.raises(Exception):
net.add_node(a)
with pytest.raises(Exception):
net.add_edge(e)
# add edges with uids
e = Edge(a, b, uid='a-b')
f = Edge(b, c, uid='b-c')
g = Edge(a, b, uid='a-b')
h = Edge(a, b, uid='ab')
net = Network()
net.add_edge(e)
net.add_edge(f)
with pytest.raises(Exception):
net.add_edge(h)
assert len(net.edges) == 2
assert len(net.nodes) == 3
with pytest.raises(Exception):
net.add_edge(g)
with pytest.raises(Exception):
net.add_edge(e)
# add edges and nodes
net = Network()
net.add_edge(e)
# add new node with same uid
with pytest.raises(Exception):
net.add_node('a')
# add same node
with pytest.raises(Exception):
net.add_node(a)
# add node and edge with the node
a1 = Node('a')
a2 = Node('a')
b = Node('b')
e1 = Edge(a2, b)
net = Network()
net.add_node(a1)
with pytest.raises(Exception):
net.add_edge(e1)
e2 = Edge(net.nodes['a'], b)
net.add_edge(e2)
# net add edge via string and nodes
net = Network()
net.add_node('a')
net.add_node('b')
net.add_edge('a', 'b')
assert len(net.nodes) == 2
assert len(net.edges) == 1
with pytest.raises(Exception):
net.add_edge('a', 'b')
net = Network(multiedges=True)
net.add_node('a')
net.add_node('b')
net.add_edge('a', 'b')
assert len(net.nodes) == 2
assert len(net.edges) == 1
net.add_edge('a', 'b')
assert len(net.nodes) == 2
assert len(net.edges) == 2
c = Node('c')
net.add_edge('b', c)
assert len(net.nodes) == 3
assert len(net.edges) == 3
a = Node('a')
with pytest.raises(Exception):
net.add_edge(a, 'b')
with pytest.raises(Exception):
net.add_edge(None)
net = Network()
net.add_edge('a', 'b', uid='a-b', length=10)
assert net.number_of_nodes() == 2
assert net.number_of_edges() == 1
assert isinstance(net.edges['a-b'], Edge)
assert net.edges['a-b'].uid == 'a-b'
assert net.edges['a-b']['length'] == 10
assert net.nodes['a'].uid == 'a'
assert net.nodes['b'].uid == 'b'
b = net.nodes['b']
c = Node('c')
net.add_edge(b, c, uid='c-d', length=5)
assert net.number_of_edges() == 2
net.add_edge('c', 'd', uid='c-2-d')
assert net.number_of_edges() == 3
assert net.edges['c-2-d'].v.uid == 'c'
net.add_edge('a', 'd', uid='a-d')
assert net.edges['a-d'].uid == 'a-d'
ab = Edge(Node('a'), Node('b'), uid='a-b')
net = Network()
net.add_edge(ab, color='blue')
assert net.edges['a-b']['color'] == 'blue'
net = Network()
net.add_node("A")
net.add_edge("A", "B")
assert net.number_of_edges() == 1
assert net.number_of_nodes() == 2
net = Network()
edges = [("A", "B"), ("B", "C")]
for edge in edges:
net.add_edge(edge)
assert net.number_of_edges() == 2
assert net.number_of_nodes() == 3