1
0
mirror of https://github.com/gsi-upm/soil synced 2024-11-14 07:22:29 +00:00
soil/benchmarks/noop/soilent_gens_pqueue.py
J. Fernando Sánchez 5e93399d58 Initial benchmarking
2023-05-03 12:07:55 +02:00

29 lines
626 B
Python

from soil import Agent, Environment
from soilent import PQueueScheduler
class NoopAgent(Agent):
num_calls = 0
def step(self):
while True:
self.num_calls += 1
# yield self.delay(1)
yield
class NoopEnvironment(Environment):
num_agents = 100
schedule_class = PQueueScheduler
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, PQueueScheduler)