def test_random_walk_from_source(self):
net = models.Network()
self.db.add(net)
self.db.commit()
agent1 = nodes.ReplicatorAgent(network=net)
agent2 = nodes.ReplicatorAgent(network=net)
agent3 = nodes.ReplicatorAgent(network=net)
agent1.connect(whom=agent2)
agent2.connect(whom=agent3)
source = nodes.RandomBinaryStringSource(network=net)
from operator import attrgetter
source.connect(
whom=min(net.nodes(type=Agent), key=attrgetter('creation_time')))
source.create_information()
processes.random_walk(net)
agent1.receive()
msg = agent1.infos()[0].contents
processes.random_walk(net)
agent2.receive()
agent2.infos()[0].contents
processes.random_walk(net)
agent3.receive()
agent3.infos()[0].contents
assert msg == agent3.infos()[0].contents
def test_create_random_binary_string_source(self):
net = models.Network()
self.add(net)
source = nodes.RandomBinaryStringSource(network=net)
self.add(source)
assert source
def test_network_repr(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
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_chain_repr(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.ReplicatorAgent(network=net)
net.add_node(agent)
self.db.commit()
assert repr(net) == "<Network-" + str(
net.id
) + "-chain with 5 nodes, 4 vectors, 0 infos, 0 transmissions and 0 transformations>"
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_moran_process_cultural(self):
# Create a fully-connected network.
net = models.Network()
self.db.add(net)
self.db.commit()
agent1 = nodes.ReplicatorAgent(network=net)
agent2 = nodes.ReplicatorAgent(network=net)
agent3 = nodes.ReplicatorAgent(network=net)
self.db.commit()
agent1.connect(whom=agent2)
agent1.connect(whom=agent3)
agent2.connect(whom=agent1)
agent2.connect(whom=agent3)
agent3.connect(whom=agent1)
agent3.connect(whom=agent2)
# Add a global source and broadcast to all the nodes.
source = nodes.RandomBinaryStringSource(network=net)
for agent in net.nodes(type=Agent):
source.connect(whom=agent)
source.transmit(to_whom=agent)
agent.receive()
# Run a Moran process for 100 steps.
for _ in xrange(100):
processes.moran_cultural(net)
for agent in net.nodes(type=Agent):
agent.receive()
# Ensure that the process had reached fixation.
from operator import attrgetter
assert max(agent1.infos(),
key=attrgetter('creation_time')).contents == max(
agent2.infos(),
key=attrgetter('creation_time')).contents
assert max(agent2.infos(),
key=attrgetter('creation_time')).contents == max(
agent3.infos(),
key=attrgetter('creation_time')).contents
assert max(agent3.infos(),
key=attrgetter('creation_time')).contents == max(
agent1.infos(),
key=attrgetter('creation_time')).contents
def test_network_add_source_local(self):
net = networks.Network()
self.db.add(net)
self.db.commit()
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_transmit_random_binary_string_source(self):
net = models.Network()
self.add(net)
source = nodes.RandomBinaryStringSource(network=net)
agent = nodes.ReplicatorAgent(network=net)
self.db.add(source)
self.db.add(agent)
self.db.commit()
source.connect(whom=agent)
self.add(source, agent)
source.transmit(to_whom=agent)
self.db.commit()
agent.receive()
self.db.commit()
assert agent.infos()[0].contents in ["00", "01", "10", "11"]
def test_broadcast_random_binary_string_source(self):
net = models.Network()
self.add(net)
source = nodes.RandomBinaryStringSource(network=net)
agent1 = nodes.ReplicatorAgent(network=net)
agent2 = nodes.ReplicatorAgent(network=net)
self.db.add(agent1)
self.db.add(agent2)
self.db.commit()
source.connect(whom=agent1)
source.connect(whom=agent2)
self.add(source, agent1, agent2)
source.transmit(what=source.create_information())
self.db.commit()
agent1.receive()
agent2.receive()
self.db.commit()
assert agent1.infos()[0].contents in ["00", "01", "10", "11"]
assert agent2.infos()[0].contents in ["00", "01", "10", "11"]
def test_moran_process_sexual(self):
# Create a fully-connected network.
net = networks.Network()
self.db.add(net)
self.db.commit()
agent1 = nodes.ReplicatorAgent(network=net)
agent2 = nodes.ReplicatorAgent(network=net)
agent3 = nodes.ReplicatorAgent(network=net)
agent1.connect(direction="both", whom=[agent2, agent3])
agent2.connect(direction="both", whom=agent3)
# Add a global source and broadcast to all the nodes.
source = nodes.RandomBinaryStringSource(network=net)
source.connect(direction="to", whom=net.nodes(type=Agent))
source.create_information()
for agent in net.nodes(type=Agent):
source.transmit(to_whom=agent)
agent.receive()
# Run a Moran process for 100 steps.
for _ in range(100):
nodes.ReplicatorAgent(network=net)
processes.moran_sexual(net)
for agent in net.nodes(type=Agent):
agent.receive()
# Ensure that the process had reached fixation.
assert agent1.failed is True
assert agent2.failed is True
assert agent3.failed is True
for a in net.nodes(type=Agent):
for a2 in net.nodes(type=Agent):
assert a.infos()[0].contents == a2.infos()[0].contents
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)