-
Notifications
You must be signed in to change notification settings - Fork 30
/
test_runner.py
59 lines (44 loc) · 2.19 KB
/
test_runner.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from sem import SimulationRunner, ParallelRunner
import os
import pytest
###################
# Runner creation #
###################
@pytest.fixture(scope='function', params=[['ParallelRunner', True],
['ParallelRunner', False]])
def runner(ns_3_compiled, ns_3_compiled_debug, config, request):
ns_3_folder = ns_3_compiled if request.param[1] is True else ns_3_compiled_debug
if request.param[0] == 'SimulationRunner':
return SimulationRunner(ns_3_folder, config['script'],
optimized=request.param[1])
elif request.param[0] == 'ParallelRunner':
return ParallelRunner(ns_3_folder, config['script'],
optimized=request.param[1])
def test_get_available_parameters(runner, config):
# Try getting the available parameters of the script
assert runner.get_available_parameters() == config['params']
@pytest.mark.parametrize('runner',
[
['SimulationRunner', True],
['ParallelRunner', True],
],
indirect=True)
def test_run_simulations(runner, config,
parameter_combination):
# Make sure that simulations run without any issue
data_dir = os.path.join(config['campaign_dir'], 'data')
list(runner.run_simulations([parameter_combination], data_dir))
def test_scratch_script(ns_3_compiled, config):
data_dir = os.path.join(config['campaign_dir'], 'data')
runner = ParallelRunner(ns_3_compiled, 'scratch-simulator')
list(runner.run_simulations([{}], data_dir))
def test_non_existent_script(ns_3_compiled):
with pytest.raises(ValueError):
ParallelRunner(ns_3_compiled, 'non_existing_script')
def test_script_without_args(ns_3_compiled):
ParallelRunner(ns_3_compiled, 'sample-random-variable')
def test_empty_param_list(ns_3_compiled, config, parameter_combination):
runner = SimulationRunner(ns_3_compiled, 'error-throwing-example')
data_dir = os.path.join(config['campaign_dir'], 'data')
with pytest.raises(Exception):
list(runner.run_simulations([parameter_combination], data_dir))