From 6357bd51859bb29629d360309b2b4134b6ae766d Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Thu, 21 Feb 2019 16:55:19 +0100 Subject: dfs: always return tuples --- bin/test_automata.py | 8 ++++---- lib/automata.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/bin/test_automata.py b/bin/test_automata.py index aafc50b..c6f473b 100755 --- a/bin/test_automata.py +++ b/bin/test_automata.py @@ -88,7 +88,7 @@ example_json_1 = { def dfs_tran_to_name(runs: list, with_args: bool) -> list: if with_args: return list(map(lambda run: list(map(lambda x: (x[0].name, x[1]), run)), runs)) - return list(map(lambda run: list(map(lambda transition: transition.name, run)), runs)) + return list(map(lambda run: list(map(lambda x: (x[0].name), run)), runs)) class TestPTA(unittest.TestCase): def test_dfs(self): @@ -105,9 +105,9 @@ class TestPTA(unittest.TestCase): pta.add_transition('UNINITIALIZED', 'IDLE', 'init') pta.add_transition('IDLE', 'IDLE', 'set1') pta.add_transition('IDLE', 'IDLE', 'set2') - self.assertEqual(list(map(lambda x: list(map(lambda y: y.name, x)), pta.dfs(0))), [['init']]) - self.assertEqual(sorted(map(lambda x: list(map(lambda y: y.name, x)), pta.dfs(1))), [['init', 'set1'], ['init', 'set2']]) - self.assertEqual(sorted(map(lambda x: list(map(lambda y: y.name, x)), pta.dfs(2))), [['init', 'set1', 'set1'], + self.assertEqual(dfs_tran_to_name(pta.dfs(0), False), [['init']]) + self.assertEqual(sorted(dfs_tran_to_name(pta.dfs(1), False)), [['init', 'set1'], ['init', 'set2']]) + self.assertEqual(sorted(dfs_tran_to_name(pta.dfs(2), False)), [['init', 'set1', 'set1'], ['init', 'set1', 'set2'], ['init', 'set2', 'set1'], ['init', 'set2', 'set2']]) diff --git a/lib/automata.py b/lib/automata.py index 640da96..f9c8ad6 100755 --- a/lib/automata.py +++ b/lib/automata.py @@ -93,7 +93,7 @@ class State: for args in zip(*trans.argument_values): yield [(trans, args)] else: - yield [trans] + yield [(trans,)] else: for trans in self.outgoing_transitions.values(): for suffix in trans.destination.dfs(depth - 1, with_arguments = with_arguments): @@ -113,7 +113,7 @@ class State: new_suffix.extend(suffix) yield new_suffix else: - new_suffix = [trans] + new_suffix = [(trans,)] new_suffix.extend(suffix) yield new_suffix -- cgit v1.2.3