from unittest import TestCase import os from os.path import join from soil import serialization, simulation ROOT = os.path.abspath(os.path.dirname(__file__)) EXAMPLES = join(ROOT, '..', 'examples') FORCE_TESTS = os.environ.get('FORCE_TESTS', '') class TestExamples(TestCase): pass def make_example_test(path, config): def wrapped(self): root = os.getcwd() for s in simulation.all_from_config(path): iterations = s.max_time * s.num_trials if iterations > 1000: s.max_time = 100 s.num_trials = 1 if config.get('skip_test', False) and not FORCE_TESTS: self.skipTest('Example ignored.') envs = s.run_simulation(dry_run=True) assert envs for env in envs: assert env try: n = config['network_params']['n'] assert len(list(env.network_agents)) == n assert env.now > 0 # It has run assert env.now <= config['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, config=config) fname = os.path.basename(path) p.__name__ = 'test_example_file_%s' % fname p.__doc__ = '%s should be a valid configuration' % fname setattr(TestExamples, p.__name__, p) del p add_example_tests()