soil/tests/test_config.py

from unittest import TestCase, skip
import os
import yaml
import copy
from os.path import join

from soil import simulation, serialization, config, network, agents, utils

ROOT = os.path.abspath(os.path.dirname(__file__))
EXAMPLES = join(ROOT, "..", "examples")

FORCE_TESTS = os.environ.get("FORCE_TESTS", "")


def isequal(a, b):
    if isinstance(a, dict):
        for (k, v) in a.items():
            if v:
                isequal(a[k], b[k])
            else:
                assert not b.get(k, None)
        return
    assert a == b


# @skip("new versions of soil do not rely on configuration files")
class TestConfig(TestCase):

    def test_torvalds_config(self):
        sim = simulation.from_config(os.path.join(ROOT, "test_config.yml"))
        MAX_STEPS = 10
        INTERVAL = 2
        assert sim.interval == INTERVAL
        assert sim.max_steps == MAX_STEPS
        envs = sim.run()
        assert len(envs) == 1
        env = envs[0]
        assert env.interval == 2
        assert env.count_agents() == 3
        assert env.now == INTERVAL * MAX_STEPS 


def make_example_test(path, cfg):
    def wrapped(self):
        root = os.getcwd()
        print(path)
        s = simulation.from_config(cfg)
        iterations = s.max_time * s.iterations
        if iterations > 1000:
            s.max_time = 100
            s.iterations = 1
        if cfg.skip_test and not FORCE_TESTS:
            self.skipTest('Example ignored.')
        envs = s.run_simulation(dump=False)
        assert envs
        for env in envs:
            assert env
            try:
                n = cfg.parameters['topology']['params']['n']
                assert len(list(env.network_agents)) == n
                assert env.now > 0  # It has run
                assert env.now <= cfg.max_time  # But not further than allowed
            except KeyError:
                pass

    return wrapped


def add_example_tests():
    for config, path in serialization.load_files(
        join(EXAMPLES, "*", "*.yml"),
        join(EXAMPLES, "*.yml"),
    ):
        p = make_example_test(path=path, cfg=config)
        fname = os.path.basename(path)
        p.__name__ = "test_example_file_%s" % fname
        p.__doc__ = "%s should be a valid configuration" % fname
        setattr(TestConfig, p.__name__, p)
        del p


add_example_tests()