mirror of
https://github.com/gsi-upm/soil
synced 2024-11-13 23:12:28 +00:00
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import os
|
|
import io
|
|
import tempfile
|
|
import shutil
|
|
from time import time
|
|
|
|
from unittest import TestCase
|
|
from soil import exporters
|
|
from soil import simulation
|
|
|
|
from soil.stats import distribution
|
|
|
|
class Dummy(exporters.Exporter):
|
|
started = False
|
|
trials = 0
|
|
ended = False
|
|
total_time = 0
|
|
called_start = 0
|
|
called_trial = 0
|
|
called_end = 0
|
|
|
|
def start(self):
|
|
self.__class__.called_start += 1
|
|
self.__class__.started = True
|
|
|
|
def trial(self, env, stats):
|
|
assert env
|
|
self.__class__.trials += 1
|
|
self.__class__.total_time += env.now
|
|
self.__class__.called_trial += 1
|
|
|
|
def end(self, stats):
|
|
self.__class__.ended = True
|
|
self.__class__.called_end += 1
|
|
|
|
|
|
class Exporters(TestCase):
|
|
def test_basic(self):
|
|
config = {
|
|
'name': 'exporter_sim',
|
|
'network_params': {},
|
|
'agent_type': 'CounterModel',
|
|
'max_time': 2,
|
|
'num_trials': 5,
|
|
'environment_params': {}
|
|
}
|
|
s = simulation.from_config(config)
|
|
for env in s.run_simulation(exporters=[Dummy], dry_run=True):
|
|
assert env.now <= 2
|
|
|
|
assert Dummy.started
|
|
assert Dummy.ended
|
|
assert Dummy.called_start == 1
|
|
assert Dummy.called_end == 1
|
|
assert Dummy.called_trial == 5
|
|
assert Dummy.trials == 5
|
|
assert Dummy.total_time == 2*5
|
|
|
|
def test_writing(self):
|
|
'''Try to write CSV, GEXF, sqlite and YAML (without dry_run)'''
|
|
n_trials = 5
|
|
config = {
|
|
'name': 'exporter_sim',
|
|
'network_params': {
|
|
'generator': 'complete_graph',
|
|
'n': 4
|
|
},
|
|
'agent_type': 'CounterModel',
|
|
'max_time': 2,
|
|
'num_trials': n_trials,
|
|
'environment_params': {}
|
|
}
|
|
output = io.StringIO()
|
|
s = simulation.from_config(config)
|
|
tmpdir = tempfile.mkdtemp()
|
|
envs = s.run_simulation(exporters=[
|
|
exporters.default,
|
|
exporters.csv,
|
|
exporters.gexf,
|
|
],
|
|
stats=[distribution,],
|
|
outdir=tmpdir,
|
|
exporter_params={'copy_to': output})
|
|
result = output.getvalue()
|
|
|
|
simdir = os.path.join(tmpdir, s.group or '', s.name)
|
|
with open(os.path.join(simdir, '{}.dumped.yml'.format(s.name))) as f:
|
|
result = f.read()
|
|
assert result
|
|
|
|
try:
|
|
for e in envs:
|
|
with open(os.path.join(simdir, '{}.gexf'.format(e.name))) as f:
|
|
result = f.read()
|
|
assert result
|
|
|
|
with open(os.path.join(simdir, '{}.csv'.format(e.name))) as f:
|
|
result = f.read()
|
|
assert result
|
|
finally:
|
|
shutil.rmtree(tmpdir)
|