mirror of
https://github.com/gsi-upm/soil
synced 2024-09-21 07:41:43 +00:00
29 lines
626 B
Python
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)
|