soil/benchmarks/noop/soilent_async_pqueue.py

from soil import Agent, Environment
from soil.time import SoilentPQueueActivation


class NoopAgent(Agent):
    num_calls = 0

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

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

    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, SoilentPQueueActivation)