def test_create_agent_generic_transmit_to_all(self):
net = models.Network()
self.db.add(net)
agent1 = nodes.Agent(network=net)
agent2 = nodes.Agent(network=net)
agent3 = nodes.Agent(network=net)
agent1.connect(direction="to", whom=agent2)
agent1.connect(direction="to", whom=agent3)
agent1.transmit(to_whom=models.Node)
def test_network_add_node(self, db_session):
net = networks.Network()
db_session.add(net)
nodes.Agent(network=net)
nodes.Agent(network=net)
nodes.Agent(network=net)
assert len(net.nodes(type=nodes.Agent)) == 3
assert len(net.vectors()) == 0
assert len(net.nodes(type=nodes.Source)) == 0
def test_network_add_node(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
nodes.Agent(network=net)
nodes.Agent(network=net)
nodes.Agent(network=net)
assert len(net.nodes(type=nodes.Agent)) == 3
assert len(net.vectors()) == 0
assert len(net.nodes(type=nodes.Source)) == 0
def test_network_add_source_local(self, db_session):
net = networks.Network()
db_session.add(net)
nodes.Agent(network=net)
nodes.Agent(network=net)
source = nodes.RandomBinaryStringSource(network=net)
source.connect(whom=net.nodes(type=nodes.Agent)[0])
assert len(net.vectors()) == 1
assert [len(n.vectors(direction="outgoing")) for n in net.nodes(type=nodes.Agent)] == [0, 0]
assert len(net.nodes(type=nodes.Source)[0].vectors(direction="outgoing")) == 1
def test_network_repr(self, db_session):
net = networks.Network()
db_session.add(net)
nodes.Agent(network=net)
nodes.Agent(network=net)
source = nodes.RandomBinaryStringSource(network=net)
source.connect(whom=net.nodes(type=nodes.Agent))
assert repr(net) == ("<Network-" + str(net.id) +
"-network with 3 nodes, 2 vectors, "
"0 infos, 0 transmissions and 0 transformations>")
def test_network_degrees(self, db_session):
net = networks.Network()
db_session.add(net)
agent1 = nodes.Agent(network=net)
agent2 = nodes.Agent(network=net)
assert [len(n.vectors(direction="outgoing")) for n in net.nodes()] == [0, 0]
agent1.connect(whom=agent2)
assert 1 in [len(n.vectors(direction="outgoing")) for n in net.nodes()]
assert 0 in [len(n.vectors(direction="outgoing")) for n in net.nodes()]
def test_network_vectors(self, db_session):
net = networks.Network()
db_session.add(net)
assert len(net.vectors()) == 0
agent1 = nodes.Agent(network=net)
agent2 = nodes.Agent(network=net)
agent1.connect(whom=agent2)
assert len(net.vectors()) == 1
assert net.vectors()[0].origin == agent1
assert net.vectors()[0].destination == agent2
def test_network_vectors(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
assert len(net.vectors()) == 0
agent1 = nodes.Agent(network=net)
agent2 = nodes.Agent(network=net)
agent1.connect(whom=agent2)
assert len(net.vectors()) == 1
assert net.vectors()[0].origin == agent1
assert net.vectors()[0].destination == agent2
def test_network_downstream_nodes(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
node1 = models.Node(network=net)
node2 = models.Node(network=net)
agent1 = nodes.Agent(network=net)
agent2 = nodes.ReplicatorAgent(network=net)
source1 = nodes.Source(network=net)
nodes.Source(network=net)
node1.connect(whom=[node2, agent1, agent2])
raises(TypeError, node1.connect, whom=source1)
assert set(node1.neighbors(direction="to")) == set(
[node2, agent1, agent2])
assert len(node1.vectors(direction="outgoing")) == 3
assert set(node1.neighbors(direction="to",
type=nodes.Agent)) == set([agent1, agent2])
agent1.fail()
agent2.fail()
raises(ValueError, node1.neighbors, direction="ghbhfgjd")
def test_network_agents(self, db_session):
net = networks.Network()
db_session.add(net)
assert len(net.nodes(type=nodes.Agent)) == 0
agent = nodes.Agent(network=net)
assert net.nodes(type=nodes.Agent) == [agent]
assert isinstance(net, models.Network)
def test_fully_connected_repr(self):
net = networks.FullyConnected()
self.db.add(net)
self.db.commit()
for i in range(4):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert repr(net) == ("<Network-" + str(net.id) +
"-fully-connected with 4 nodes, 12 vectors, "
"0 infos, 0 transmissions and 0 transformations>")
def test_network_agents(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
assert len(net.nodes(type=nodes.Agent)) == 0
agent = nodes.Agent(network=net)
assert net.nodes(type=nodes.Agent) == [agent]
assert isinstance(net, models.Network)
def test_network_nodes(self, db_session):
net = models.Network()
db_session.add(net)
node1 = models.Node(network=net)
node2 = models.Node(network=net)
agent1 = nodes.Agent(network=net)
agent2 = nodes.Agent(network=net)
agent3 = nodes.Agent(network=net)
assert set([node1, node2, agent1, agent2, agent3]) == set(net.nodes())
assert set([agent1, agent2, agent3]) == set(net.nodes(type=nodes.Agent))
node1.fail()
agent1.fail()
assert set(net.nodes()) == set([node2, agent2, agent3])
assert set(net.nodes(failed="all")) == set([node1, node2, agent1, agent2, agent3])
assert set(net.nodes(failed=True)) == set([node1, agent1])
assert set(net.nodes(type=nodes.Agent, failed="all")) == set([agent1, agent2, agent3])
def test_fail_agent(self):
net = models.Network()
self.db.add(net)
agent = nodes.Agent(network=net)
self.db.commit()
assert agent.failed is False
assert agent.time_of_death is None
agent.fail()
assert agent.failed is True
assert agent.time_of_death is not None
def test_network_add_source_global(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
agent1 = nodes.Agent(network=net)
nodes.Agent(network=net)
# self.db.commit()
source = nodes.RandomBinaryStringSource(network=net)
source.connect(whom=net.nodes(type=nodes.Agent))
assert len(net.vectors()) == 2
assert source.network == net
assert agent1.network == net
assert [
len(n.vectors(direction="outgoing"))
for n in net.nodes(type=nodes.Agent)
] == [0, 0]
assert len(
net.nodes(type=nodes.Source)[0].vectors(direction="outgoing")) == 2
def test_scale_free_repr(self):
net = networks.ScaleFree(m0=4, m=4)
self.db.add(net)
self.db.commit()
for i in range(6):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert repr(net) == ("<Network-" + str(net.id) +
"-scale-free with 6 nodes, 28 vectors, "
"0 infos, 0 transmissions and 0 transformations>")
def test_create_empty(self):
"""Empty networks should have nodes, but no edges."""
net = networks.Empty()
self.db.add(net)
self.db.commit()
for i in range(10):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert len(net.nodes(type=nodes.Agent)) == 10
assert len(net.vectors()) == 0
def test_create_scale_free(self):
m0 = 4
m = 4
net = networks.ScaleFree(m0=m0, m=m)
self.db.add(net)
self.db.commit()
for i in range(m0):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert len(net.nodes(type=nodes.Agent)) == m0
assert len(net.vectors()) == m0 * (m0 - 1)
agent1 = nodes.Agent(network=net)
net.add_node(agent1)
assert len(net.nodes(type=nodes.Agent)) == m0 + 1
assert len(net.vectors()) == m0 * (m0 - 1) + 2 * m
agent2 = nodes.Agent(network=net)
net.add_node(agent2)
assert len(net.nodes(type=nodes.Agent)) == m0 + 2
assert len(net.vectors()) == m0 * (m0 - 1) + 2 * 2 * m
def test_create_fully_connected(self):
net = networks.FullyConnected()
self.db.add(net)
self.db.commit()
for i in range(4):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert len(net.nodes(type=nodes.Agent)) == 4
assert len(net.vectors()) == 12
assert [
len(n.vectors(direction="outgoing"))
for n in net.nodes(type=nodes.Agent)
] == [3, 3, 3, 3]
def test_create_empty_with_source(self):
"""A sourced empty network should have nodes and an edge for each."""
net = networks.Empty()
self.db.add(net)
self.db.commit()
for i in range(10):
agent = nodes.Agent(network=net)
net.add_node(agent)
source = nodes.Source(network=net)
net.add_source(source)
assert len(net.nodes(type=nodes.Agent)) == 10
assert len(net.vectors()) == 10
def test_node_failure(self, db_session):
net = networks.Network()
db_session.add(net)
for _ in range(5):
nodes.Agent(network=net)
nodes.Source(network=net)
assert len(net.nodes(type=nodes.Agent)) == 5
random.choice(net.nodes(type=nodes.Agent)).fail()
assert len(net.nodes(type=nodes.Agent)) == 4
assert len(net.nodes(type=nodes.Agent, failed="all")) == 5
assert len(net.nodes()) == 5
assert len(net.nodes(failed="all")) == 6
assert len(net.nodes(failed=True)) == 1
def test_create_chain(self):
net = networks.Chain()
self.db.add(net)
self.db.commit()
source = nodes.RandomBinaryStringSource(network=net)
net.add_node(source)
for i in range(4):
agent = nodes.Agent(network=net)
net.add_node(agent)
assert len(net.nodes(type=nodes.Agent)) == 4
assert len(net.nodes(type=nodes.Source)) == 1
assert len(net.vectors()) == 4
assert net.nodes(type=nodes.Agent)[0].network == net
assert net.nodes(type=nodes.Source)[0].network == net
def test_create_sequential_microsociety(self):
"""Create a sequential microsociety."""
net = networks.SequentialMicrosociety(n=3)
self.db.add(net)
self.db.commit()
net.add_node(nodes.RandomBinaryStringSource(network=net))
agent1 = nodes.Agent(network=net)
net.add_node(agent1)
agent2 = nodes.Agent(network=net)
net.add_node(agent2)
agent3 = nodes.Agent(network=net)
net.add_node(agent3)
agent4 = nodes.Agent(network=net)
net.add_node(agent4)
agent5 = nodes.Agent(network=net)
net.add_node(agent5)
agent6 = nodes.Agent(network=net)
net.add_node(agent6)
assert len(agent1.vectors(direction="outgoing")) == 2
assert len(agent2.vectors(direction="outgoing")) == 2
assert len(agent3.vectors(direction="outgoing")) == 2
assert agent2.is_connected(direction="to", whom=agent3)
assert agent2.is_connected(direction="to", whom=agent4)
assert not agent2.is_connected(direction="to", whom=agent5)
assert agent3.is_connected(direction="to", whom=agent4)
assert agent3.is_connected(direction="to", whom=agent5)
assert not agent3.is_connected(direction="to", whom=agent6)
def test_not_full_if_nodes_fewer_than_max_size(self, a):
net = a.network(max_size=2)
nodes.Agent(network=net)
assert not net.full
def test_full_if_nodes_equal_max_size(self, a):
net = a.network(max_size=1)
nodes.Agent(network=net)
assert net.full
def test_create_agent_generic(self):
net = models.Network()
self.db.add(net)
agent = nodes.Agent(network=net)
assert agent
