mirror of
https://github.com/gsi-upm/soil
synced 2025-06-30 18:52:21 +00:00
73282530fd
All test pass, except for the TestConfig suite, which is not too critical as the plan for this version onwards is to avoid configuration as much as possible.
47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
from . import BaseAgent, NetworkAgent
|
|
|
|
|
|
class Ticker(BaseAgent):
|
|
times = 0
|
|
|
|
def step(self):
|
|
self.times += 1
|
|
|
|
class CounterModel(NetworkAgent):
|
|
"""
|
|
Dummy behaviour. It counts the number of nodes in the network and neighbors
|
|
in each step and adds it to its state.
|
|
"""
|
|
|
|
times = 0
|
|
neighbors = 0
|
|
total = 0
|
|
|
|
def step(self):
|
|
# Outside effects
|
|
total = len(list(self.model.schedule._agents))
|
|
neighbors = len(list(self.get_neighbors()))
|
|
self["times"] = self.get("times", 0) + 1
|
|
self["neighbors"] = neighbors
|
|
self["total"] = total
|
|
|
|
|
|
class AggregatedCounter(NetworkAgent):
|
|
"""
|
|
Dummy behaviour. It counts the number of nodes in the network and neighbors
|
|
in each step and adds it to its state.
|
|
"""
|
|
|
|
times = 0
|
|
neighbors = 0
|
|
total = 0
|
|
|
|
def step(self):
|
|
# Outside effects
|
|
self["times"] += 1
|
|
neighbors = len(list(self.get_neighbors()))
|
|
self["neighbors"] += neighbors
|
|
total = len(list(self.model.schedule.agents))
|
|
self["total"] += total
|
|
self.debug("Running for step: {}. Total: {}".format(self.now, total))
|