soil/examples/random_delays/random_delays.py

"""
Example of setting a 
Example of a fully programmatic simulation, without definition files.
"""
from soil import Simulation, agents
from soil.time import Delta


class MyAgent(agents.FSM):
    """
    An agent that first does a ping
    """

    defaults = {"pong_counts": 2}

    @agents.default_state
    @agents.state
    def ping(self):
        self.info("Ping")
        return self.pong, Delta(self.random.expovariate(1 / 16))

    @agents.state
    def pong(self):
        self.info("Pong")
        self.pong_counts -= 1
        self.info(str(self.pong_counts))
        if self.pong_counts < 1:
            return self.die()
        return None, Delta(self.random.expovariate(1 / 16))


s = Simulation(
    name="Programmatic",
    network_agents=[{"agent_class": MyAgent, "id": 0}],
    topology={"nodes": [{"id": 0}], "links": []},
    num_trials=1,
    max_time=100,
    agent_class=MyAgent,
    dry_run=True,
)


envs = s.run()