diff options
author | Daniel Friesel <derf@finalrewind.org> | 2018-04-24 10:46:32 +0200 |
---|---|---|
committer | Daniel Friesel <derf@finalrewind.org> | 2018-04-24 10:46:32 +0200 |
commit | d3ad69999ddabd7540ab96a92f56300803f608f5 (patch) | |
tree | 578fc8a8c3ada65bd6548a1996b72db374f8aac0 /bin | |
parent | 4384b92ec6dd950559afcca59bca4c8f3c41eec2 (diff) |
PTA: Use parameter dictionary internally
Diffstat (limited to 'bin')
-rwxr-xr-x | bin/test_automata.py | 265 |
1 files changed, 265 insertions, 0 deletions
diff --git a/bin/test_automata.py b/bin/test_automata.py new file mode 100755 index 0000000..5a62931 --- /dev/null +++ b/bin/test_automata.py @@ -0,0 +1,265 @@ +#!/usr/bin/env python3 + +from automata import PTA +import unittest + +class TestPTA(unittest.TestCase): + def test_dfs(self): + pta = PTA(['IDLE', 'TX']) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init') + pta.add_transition('IDLE', 'TX', 'send') + pta.add_transition('TX', 'IDLE', 'txComplete') + self.assertEqual(list(pta.dfs(0)), [['init']]) + self.assertEqual(list(pta.dfs(1)), [['init', 'send']]) + self.assertEqual(list(pta.dfs(2)), [['init', 'send', 'txComplete']]) + self.assertEqual(list(pta.dfs(3)), [['init', 'send', 'txComplete', 'send']]) + + pta = PTA(['IDLE']) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init') + pta.add_transition('IDLE', 'IDLE', 'set1') + pta.add_transition('IDLE', 'IDLE', 'set2') + self.assertEqual(list(pta.dfs(0)), [['init']]) + self.assertEqual(list(pta.dfs(1)), [['init', 'set1'], ['init', 'set2']]) + self.assertEqual(list(pta.dfs(2)), [['init', 'set1', 'set1'], + ['init', 'set1', 'set2'], + ['init', 'set2', 'set1'], + ['init', 'set2', 'set2']]) + + def test_from_json(self): + json_input = { + 'parameters' : ['datarate', 'txbytes', 'txpower'], + 'initial_param_values' : [None, None], + 'states' : { + 'IDLE' : { + 'power' : { + 'static' : 5, + } + }, + 'TX' : { + 'power' : { + 'static' : 100, + 'function' : { + 'raw' : '0 + regression_arg(0) + regression_arg(1)' + ' * parameter(txpower) + regression_arg(2)' + ' * 1 / parameter(datarate) + regression_arg(3)' + ' * parameter(txpower) * 1 / parameter(datarate)', + 'regression_args' : [ + 13918.0985307854, + 245.185518812381, + 8368807.4712942, + 271527.656040595 + ], + }, + } + }, + }, + 'transitions' : [ + { + 'name' : 'init', + 'origin' : 'UNINITIALIZED', + 'destination' : 'IDLE', + 'duration' : { + 'static' : 50000, + }, + }, + { + 'name' : 'setTxPower', + 'origin' : 'IDLE', + 'destination' : 'IDLE', + 'duration' : { 'static' : 120 }, + 'energy ' : { 'static' : 10000 }, + 'arg_to_param_map' : [ 'txpower' ], + }, + { + 'name' : 'send', + 'origin' : 'IDLE', + 'destination' : 'TX', + 'duration' : { + 'static' : 10, + 'function' : { + 'raw' : 'regression_arg(0) + regression_arg(1)' + ' * function_arg(1)', + 'regression_args' : [48, 8], + }, + }, + 'energy' : { + 'static' : 3, + 'function' : { + 'raw' : 'regression_arg(0) + regression_arg(1)' + ' * function_arg(1)', + 'regression_args' : [3, 5], + }, + }, + 'arg_to_param_map' : [ None, 'txbytes' ], + }, + { + 'name' : 'txComplete', + 'origin' : 'TX', + 'destination' : 'IDLE', + 'is_interrupt' : 1, + 'timeout' : { + 'static' : 2000, + 'function' : { + 'raw' : 'regression_arg(0) + regression_arg(1)' + ' * parameter(txbytes)', + 'regression_args' : [ 500, 16 ], + }, + }, + } + ], + } + + def test_simulation(self): + pta = PTA() + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', duration = 50000) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True) + trace = [ + ['init'], + ['sleep', 10000000], + ['send', 'foo', 3], + ['sleep', 5000000], + ['send', 'foo', 3] + ] + expected_energy = 5. * 10000000 + 3 + 100 * 2000 + 5 * 5000000 + 3 + 100 * 2000 + expected_duration = 50000 + 10000000 + 10 + 2000 + 5000000 + 10 + 2000 + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, {}) + + def test_simulation_param_none(self): + pta = PTA(parameters = ['txpower', 'length']) + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', energy = 500000, duration = 50000) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True) + trace = [ + ['init'], + ] + expected_energy = 500000 + expected_duration = 50000 + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, { + 'txpower' : None, + 'length' : None + }) + + def test_simulation_param_set(self): + pta = PTA(parameters = ['txpower', 'length']) + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', energy = 500000, duration = 50000) + pta.add_transition('IDLE', 'IDLE', 'setTxPower', energy = 10000, duration = 120, + param_update_function = lambda param, arg: {**param, 'txpower' : arg[0]}) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True) + trace = [ + ['init'], + ['setTxPower', 10] + ] + expected_energy = 510000 + expected_duration = 50120 + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, { + 'txpower' : 10, + 'length' : None + }) + + def test_simulation_arg_function(self): + pta = PTA(parameters = ['txpower', 'length']) + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', energy = 500000, duration = 50000) + pta.add_transition('IDLE', 'IDLE', 'setTxPower', energy = 10000, duration = 120, + param_update_function = lambda param, arg: {**param, 'txpower' : arg[0]}) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10, + energy_function = lambda param, arg: 3 + 5 * arg[1], + duration_function = lambda param, arg: 48 + 8 * arg[1]) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True) + trace = [ + ['init'], + ['setTxPower', 10], + ['send', 'foo', 3], + ] + expected_energy = 500000 + 10000 + (3 + 5 * 3) + (2000 * 100) + expected_duration = 50000 + 120 + (48 + 8 * 3) + 2000 + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, { + 'txpower' : 10, + 'length' : None + }) + + pta = PTA(parameters = ['txpower', 'length']) + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', energy = 500000, duration = 50000) + pta.add_transition('IDLE', 'IDLE', 'setTxPower', energy = 10000, duration = 120, + param_update_function = lambda param, arg: {**param, 'txpower' : arg[0]}) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10, + energy_function = lambda param, arg: 3 + 5 * arg[1], + duration_function = lambda param, arg: 48 + 8 * arg[1]) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True) + trace = [ + ['init'], + ['setTxPower', 10], + ['send', 'foobar', 6], + ] + expected_energy = 500000 + 10000 + (3 + 5 * 6) + (2000 * 100) + expected_duration = 50000 + 120 + (48 + 8 * 6) + 2000 + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, { + 'txpower' : 10, + 'length' : None + }) + + + def test_simulation_param_function(self): + pta = PTA(parameters = ['length', 'txpower']) + pta.add_state('IDLE', power = 5) + pta.add_state('TX', power = 100, + power_function = lambda param, arg: 1000 + 2 * param[1]) + pta.add_transition('UNINITIALIZED', 'IDLE', 'init', energy = 500000, duration = 50000) + pta.add_transition('IDLE', 'IDLE', 'setTxPower', energy = 10000, duration = 120, + param_update_function = lambda param, arg: {**param, 'txpower' : arg[0]}) + pta.add_transition('IDLE', 'TX', 'send', energy = 3, duration = 10, + energy_function = lambda param, arg: 3 + 5 * arg[1], + param_update_function = lambda param, arg: {**param, 'length' : arg[1]}) + pta.add_transition('TX', 'IDLE', 'txComplete', timeout = 2000, is_interrupt = True, + timeout_function = lambda param, arg: 500 + 16 * param[0]) + trace = [ + ['init'], + ['setTxPower', 10], + ['send', 'foo', 3], + ] + expected_energy = 500000 + 10000 + (3 + 5 * 3) + (1000 + 2 * 10) * (500 + 16 * 3) + expected_duration = 50000 + 120 + 10 + (500 + 16 * 3) + power, duration, state, parameters = pta.simulate(trace) + self.assertEqual(power, expected_energy) + self.assertEqual(duration, expected_duration) + self.assertEqual(state.name, 'IDLE') + self.assertEqual(parameters, { + 'txpower' : 10, + 'length' : 3 + }) + + + +if __name__ == '__main__': + unittest.main() |