diff options
| author | Jenkins <jenkins@review.openstack.org> | 2015-10-15 04:27:42 +0000 |
|---|---|---|
| committer | Gerrit Code Review <review@openstack.org> | 2015-10-15 04:27:42 +0000 |
| commit | 4677fdb30e801d7f40344f7ca83cfc6e20c30160 (patch) | |
| tree | 0fafc64247d3bd0dab9d6236dd2cc2fd739de5ad /taskflow/tests/unit | |
| parent | 2d8b9dad24e226626270cf107ca8050ae82f3fdc (diff) | |
| parent | 75517eb0e01c876894ddfd60d6a8f3441de4c38a (diff) | |
| download | taskflow-4677fdb30e801d7f40344f7ca83cfc6e20c30160.tar.gz | |
Merge "Use the node built-in 'dfs_iter' instead of recursion"
Diffstat (limited to 'taskflow/tests/unit')
| -rw-r--r-- | taskflow/tests/unit/test_types.py | 24 |
1 files changed, 19 insertions, 5 deletions
diff --git a/taskflow/tests/unit/test_types.py b/taskflow/tests/unit/test_types.py index 7904492..9399c89 100644 --- a/taskflow/tests/unit/test_types.py +++ b/taskflow/tests/unit/test_types.py @@ -467,24 +467,38 @@ CEO self.assertEqual(set(['animal', 'reptile', 'mammal', 'horse', 'primate', 'monkey', 'human']), set(things)) - def test_dfs_itr_order(self): + def test_dfs_itr_left_to_right(self): + root = self._make_species() + it = root.dfs_iter(include_self=False, right_to_left=False) + things = list([n.item for n in it]) + self.assertEqual(['reptile', 'mammal', 'primate', + 'human', 'monkey', 'horse'], things) + + def test_dfs_itr_no_self(self): root = self._make_species() - things = list([n.item for n in root.dfs_iter(include_self=True)]) - self.assertEqual(['animal', 'mammal', 'horse', 'primate', - 'monkey', 'human', 'reptile'], things) things = list([n.item for n in root.dfs_iter(include_self=False)]) self.assertEqual(['mammal', 'horse', 'primate', 'monkey', 'human', 'reptile'], things) - def test_bfs_iter(self): + def test_bfs_itr(self): root = self._make_species() things = list([n.item for n in root.bfs_iter(include_self=True)]) self.assertEqual(['animal', 'reptile', 'mammal', 'primate', 'horse', 'human', 'monkey'], things) + + def test_bfs_itr_no_self(self): + root = self._make_species() things = list([n.item for n in root.bfs_iter(include_self=False)]) self.assertEqual(['reptile', 'mammal', 'primate', 'horse', 'human', 'monkey'], things) + def test_bfs_itr_right_to_left(self): + root = self._make_species() + it = root.bfs_iter(include_self=False, right_to_left=True) + things = list([n.item for n in it]) + self.assertEqual(['mammal', 'reptile', 'horse', + 'primate', 'monkey', 'human'], things) + class OrderedSetTest(test.TestCase): |