mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 15:32:29 +00:00
25 lines
569 B
Python
25 lines
569 B
Python
|
from soil import BaseAgent, Environment, Simulation
|
||
|
|
from soilent import PQueueScheduler
|
||
|
|
|
||
|
|
|
||
|
|
class NoopAgent(BaseAgent):
|
||
|
|
num_calls = 0
|
||
|
|
|
||
|
|
def step(self):
|
||
|
|
self.num_calls += 1
|
||
|
|
|
||
|
|
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)
|