from networkx import Graph import random import networkx as nx from soil import Simulation, Environment, CounterModel, parameters def mygenerator(n=5, n_edges=5): """ Just a simple generator that creates a network with n nodes and n_edges edges. Edges are assigned randomly, only avoiding self loops. """ G = nx.Graph() for i in range(n): G.add_node(i) for i in range(n_edges): nodes = list(G.nodes) n_in = random.choice(nodes) nodes.remove(n_in) # Avoid loops n_out = random.choice(nodes) G.add_edge(n_in, n_out) return G class GeneratorEnv(Environment): """Using a custom generator for the network""" generator: parameters.function = mygenerator def init(self): self.create_network(network_generator=self.generator, n=10, n_edges=5) self.init_agents(CounterModel) sim = Simulation(model=GeneratorEnv, max_steps=10, interval=1) if __name__ == '__main__': sim.run(dry_run=True)