soil/benchmarks/noop/soilent_async.py

from soil import Agent, Environment, Simulation
from soil.time import SoilentActivation


class NoopAgent(Agent):
    num_calls = 0

    async def step(self):
        while True:
            self.num_calls += 1
            # yield self.delay(1)
            await self.delay()


class NoopEnvironment(Environment):
    num_agents = 100
    schedule_class = SoilentActivation

    def init(self):
        self.add_agents(NoopAgent, k=self.num_agents)
        self.add_agent_reporter("num_calls")


if __name__ == "__main__":
    from _config import *

    res = run_sim(model=NoopEnvironment)
    for r in res:
        assert isinstance(r.schedule, SoilentActivation)