taskflow 5.12.0__py3-none-any.whl → 6.0.0__py3-none-any.whl

taskflow/tests/unit/patterns/test_graph_flow.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2014 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -76,8 +74,8 @@ class GraphFlowTest(test.TestCase):
         self.assertEqual(1, len(f))
         self.assertEqual([task], list(f))
         self.assertEqual([], list(f.iter_links()))
-        self.assertEqual(set(['a', 'b']), f.requires)
-        self.assertEqual(set(['c', 'd']), f.provides)
+        self.assertEqual({'a', 'b'}, f.requires)
+        self.assertEqual({'c', 'd'}, f.provides)
 
     def test_graph_flow_two_independent_tasks(self):
         task1 = _task(name='task1')
@@ -95,11 +93,11 @@ class GraphFlowTest(test.TestCase):
 
         self.assertEqual(2, len(f))
         self.assertCountEqual(f, [task1, task2])
-        self.assertEqual([(task1, task2, {'reasons': set(['a'])})],
+        self.assertEqual([(task1, task2, {'reasons': {'a'}})],
                          list(f.iter_links()))
 
         self.assertEqual(set(), f.requires)
-        self.assertEqual(set(['a']), f.provides)
+        self.assertEqual({'a'}, f.provides)
 
     def test_graph_flow_two_dependent_tasks_two_different_calls(self):
         task1 = _task(name='task1', provides=['a'])
@@ -108,7 +106,7 @@ class GraphFlowTest(test.TestCase):
 
         self.assertEqual(2, len(f))
         self.assertCountEqual(f, [task1, task2])
-        self.assertEqual([(task1, task2, {'reasons': set(['a'])})],
+        self.assertEqual([(task1, task2, {'reasons': {'a'}})],
                          list(f.iter_links()))
 
     def test_graph_flow_two_task_same_provide(self):
@@ -116,14 +114,14 @@ class GraphFlowTest(test.TestCase):
         task2 = _task(name='task2', provides=['a', 'c'])
         f = gf.Flow('test')
         f.add(task2, task1)
-        self.assertEqual(set(['a', 'b', 'c']), f.provides)
+        self.assertEqual({'a', 'b', 'c'}, f.provides)
 
     def test_graph_flow_ambiguous_provides(self):
         task1 = _task(name='task1', provides=['a', 'b'])
         task2 = _task(name='task2', provides=['a'])
         f = gf.Flow('test')
         f.add(task1, task2)
-        self.assertEqual(set(['a', 'b']), f.provides)
+        self.assertEqual({'a', 'b'}, f.provides)
         task3 = _task(name='task3', requires=['a'])
         self.assertRaises(exc.AmbiguousDependency, f.add, task3)
 
@@ -132,7 +130,7 @@ class GraphFlowTest(test.TestCase):
         task2 = _task(name='task2', requires=['a', 'b'])
         f = gf.Flow('test')
         f.add(task1, task2, resolve_requires=False)
-        self.assertEqual(set(['a', 'b']), f.requires)
+        self.assertEqual({'a', 'b'}, f.requires)
 
     def test_graph_flow_no_resolve_existing(self):
         task1 = _task(name='task1', requires=['a', 'b'])
@@ -140,7 +138,7 @@ class GraphFlowTest(test.TestCase):
         f = gf.Flow('test')
         f.add(task1)
         f.add(task2, resolve_existing=False)
-        self.assertEqual(set(['a', 'b']), f.requires)
+        self.assertEqual({'a', 'b'}, f.requires)
 
     def test_graph_flow_resolve_existing(self):
         task1 = _task(name='task1', requires=['a', 'b'])
@@ -148,7 +146,7 @@ class GraphFlowTest(test.TestCase):
         f = gf.Flow('test')
         f.add(task1)
         f.add(task2, resolve_existing=True)
-        self.assertEqual(set([]), f.requires)
+        self.assertEqual(set(), f.requires)
 
     def test_graph_flow_with_retry(self):
         ret = retry.AlwaysRevert(requires=['a'], provides=['b'])
@@ -156,11 +154,11 @@ class GraphFlowTest(test.TestCase):
         self.assertIs(f.retry, ret)
         self.assertEqual('test_retry', ret.name)
 
-        self.assertEqual(set(['a']), f.requires)
-        self.assertEqual(set(['b']), f.provides)
+        self.assertEqual({'a'}, f.requires)
+        self.assertEqual({'b'}, f.provides)
 
     def test_graph_flow_ordering(self):
-        task1 = _task('task1', provides=set(['a', 'b']))
+        task1 = _task('task1', provides={'a', 'b'})
         task2 = _task('task2', provides=['c'], requires=['a', 'b'])
         task3 = _task('task3', provides=[], requires=['c'])
         f = gf.Flow('test').add(task1, task2, task3)
@@ -168,8 +166,8 @@ class GraphFlowTest(test.TestCase):
         self.assertEqual(3, len(f))
 
         self.assertCountEqual(list(f.iter_links()), [
-            (task1, task2, {'reasons': set(['a', 'b'])}),
-            (task2, task3, {'reasons': set(['c'])})
+            (task1, task2, {'reasons': {'a', 'b'}}),
+            (task2, task3, {'reasons': {'c'}})
         ])
 
     def test_graph_flow_links(self):
@@ -190,7 +188,7 @@ class GraphFlowTest(test.TestCase):
         self.assertIs(linked, f)
         expected_meta = {
             'manual': True,
-            'reasons': set(['a'])
+            'reasons': {'a'}
         }
         self.assertCountEqual(list(f.iter_links()), [
             (task1, task2, expected_meta)
@@ -236,7 +234,7 @@ class GraphFlowTest(test.TestCase):
         task3 = _task('task3', provides=['c'])
         f1 = gf.Flow('nested')
         f1.add(task3)
-        tasks = set([task1, task2, f1])
+        tasks = {task1, task2, f1}
         f = gf.Flow('test').add(task1, task2, f1)
         for (n, data) in f.iter_nodes():
             self.assertIn(n, tasks)
@@ -248,7 +246,7 @@ class GraphFlowTest(test.TestCase):
         task3 = _task('task3')
         f1 = gf.Flow('nested')
         f1.add(task3)
-        tasks = set([task1, task2, f1])
+        tasks = {task1, task2, f1}
         f = gf.Flow('test').add(task1, task2, f1)
         for (u, v, data) in f.iter_links():
             self.assertIn(u, tasks)

taskflow/tests/unit/patterns/test_linear_flow.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2014 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -65,8 +63,8 @@ class LinearFlowTest(test.TestCase):
         self.assertEqual(1, len(f))
         self.assertEqual([task], list(f))
         self.assertEqual([], list(f.iter_links()))
-        self.assertEqual(set(['a', 'b']), f.requires)
-        self.assertEqual(set(['c', 'd']), f.provides)
+        self.assertEqual({'a', 'b'}, f.requires)
+        self.assertEqual({'c', 'd'}, f.provides)
 
     def test_linear_flow_two_independent_tasks(self):
         task1 = _task(name='task1')
@@ -89,7 +87,7 @@ class LinearFlowTest(test.TestCase):
                          list(f.iter_links()))
 
         self.assertEqual(set(), f.requires)
-        self.assertEqual(set(['a']), f.provides)
+        self.assertEqual({'a'}, f.provides)
 
     def test_linear_flow_two_dependent_tasks_two_different_calls(self):
         task1 = _task(name='task1', provides=['a'])
@@ -120,15 +118,15 @@ class LinearFlowTest(test.TestCase):
         self.assertIs(f.retry, ret)
         self.assertEqual('test_retry', ret.name)
 
-        self.assertEqual(set(['a']), f.requires)
-        self.assertEqual(set(['b']), f.provides)
+        self.assertEqual({'a'}, f.requires)
+        self.assertEqual({'b'}, f.provides)
 
     def test_iter_nodes(self):
         task1 = _task(name='task1')
         task2 = _task(name='task2')
         task3 = _task(name='task3')
         f = lf.Flow('test').add(task1, task2, task3)
-        tasks = set([task1, task2, task3])
+        tasks = {task1, task2, task3}
         for (node, data) in f.iter_nodes():
             self.assertIn(node, tasks)
             self.assertDictEqual({}, data)
@@ -138,7 +136,7 @@ class LinearFlowTest(test.TestCase):
         task2 = _task(name='task2')
         task3 = _task(name='task3')
         f = lf.Flow('test').add(task1, task2, task3)
-        tasks = set([task1, task2, task3])
+        tasks = {task1, task2, task3}
         for (u, v, data) in f.iter_links():
             self.assertIn(u, tasks)
             self.assertIn(v, tasks)

taskflow/tests/unit/patterns/test_unordered_flow.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2014 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -65,8 +63,8 @@ class UnorderedFlowTest(test.TestCase):
         self.assertEqual(1, len(f))
         self.assertEqual([task], list(f))
         self.assertEqual([], list(f.iter_links()))
-        self.assertEqual(set(['a', 'b']), f.requires)
-        self.assertEqual(set(['c', 'd']), f.provides)
+        self.assertEqual({'a', 'b'}, f.requires)
+        self.assertEqual({'c', 'd'}, f.provides)
 
     def test_unordered_flow_two_tasks(self):
         task1 = _task(name='task1')
@@ -74,7 +72,7 @@ class UnorderedFlowTest(test.TestCase):
         f = uf.Flow('test').add(task1, task2)
 
         self.assertEqual(2, len(f))
-        self.assertEqual(set([task1, task2]), set(f))
+        self.assertEqual({task1, task2}, set(f))
         self.assertEqual([], list(f.iter_links()))
 
     def test_unordered_flow_two_tasks_two_different_calls(self):
@@ -83,16 +81,16 @@ class UnorderedFlowTest(test.TestCase):
         f = uf.Flow('test').add(task1)
         f.add(task2)
         self.assertEqual(2, len(f))
-        self.assertEqual(set(['a']), f.requires)
-        self.assertEqual(set(['a']), f.provides)
+        self.assertEqual({'a'}, f.requires)
+        self.assertEqual({'a'}, f.provides)
 
     def test_unordered_flow_two_tasks_reverse_order(self):
         task1 = _task(name='task1', provides=['a'])
         task2 = _task(name='task2', requires=['a'])
         f = uf.Flow('test').add(task2).add(task1)
         self.assertEqual(2, len(f))
-        self.assertEqual(set(['a']), f.requires)
-        self.assertEqual(set(['a']), f.provides)
+        self.assertEqual({'a'}, f.requires)
+        self.assertEqual({'a'}, f.provides)
 
     def test_unordered_flow_two_task_same_provide(self):
         task1 = _task(name='task1', provides=['a', 'b'])
@@ -107,8 +105,8 @@ class UnorderedFlowTest(test.TestCase):
         self.assertIs(f.retry, ret)
         self.assertEqual('test_retry', ret.name)
 
-        self.assertEqual(set(['a']), f.requires)
-        self.assertEqual(set(['b']), f.provides)
+        self.assertEqual({'a'}, f.requires)
+        self.assertEqual({'b'}, f.provides)
 
     def test_unordered_flow_with_retry_fully_satisfies(self):
         ret = retry.AlwaysRevert(provides=['b', 'a'])
@@ -116,13 +114,13 @@ class UnorderedFlowTest(test.TestCase):
         f.add(_task(name='task1', requires=['a']))
         self.assertIs(f.retry, ret)
         self.assertEqual('test_retry', ret.name)
-        self.assertEqual(set([]), f.requires)
-        self.assertEqual(set(['b', 'a']), f.provides)
+        self.assertEqual(set(), f.requires)
+        self.assertEqual({'b', 'a'}, f.provides)
 
     def test_iter_nodes(self):
         task1 = _task(name='task1', provides=['a', 'b'])
         task2 = _task(name='task2', provides=['a', 'c'])
-        tasks = set([task1, task2])
+        tasks = {task1, task2}
         f = uf.Flow('test')
         f.add(task2, task1)
         for (node, data) in f.iter_nodes():

taskflow/tests/unit/persistence/base.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Rackspace Hosting All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -24,7 +22,7 @@ from taskflow import states
 from taskflow.types import failure
 
 
-class PersistenceTestMixin(object):
+class PersistenceTestMixin:
     def _get_connection(self):
         raise NotImplementedError('_get_connection() implementation required')
 
@@ -73,7 +71,7 @@ class PersistenceTestMixin(object):
         lb_ids = {}
         for i in range(0, 10):
             lb_id = uuidutils.generate_uuid()
-            lb_name = 'lb-%s-%s' % (i, lb_id)
+            lb_name = 'lb-{}-{}'.format(i, lb_id)
             lb = models.LogBook(name=lb_name, uuid=lb_id)
             lb_ids[lb_id] = True
 

taskflow/tests/unit/persistence/test_dir_persistence.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Rackspace Hosting All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -45,7 +43,7 @@ class DirPersistenceTest(testscenarios.TestWithScenarios,
         return self.backend.get_connection()
 
     def setUp(self):
-        super(DirPersistenceTest, self).setUp()
+        super().setUp()
         self.path = tempfile.mkdtemp()
         self.backend = impl_dir.DirBackend({
             'path': self.path,
@@ -55,7 +53,7 @@ class DirPersistenceTest(testscenarios.TestWithScenarios,
             conn.upgrade()
 
     def tearDown(self):
-        super(DirPersistenceTest, self).tearDown()
+        super().tearDown()
         if self.path and os.path.isdir(self.path):
             shutil.rmtree(self.path)
         self.path = None

taskflow/tests/unit/persistence/test_memory_persistence.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Rackspace Hosting All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -25,7 +23,7 @@ from taskflow.tests.unit.persistence import base
 
 class MemoryPersistenceTest(test.TestCase, base.PersistenceTestMixin):
     def setUp(self):
-        super(MemoryPersistenceTest, self).setUp()
+        super().setUp()
         self._backend = impl_memory.MemoryBackend({})
 
     def _get_connection(self):
@@ -35,7 +33,7 @@ class MemoryPersistenceTest(test.TestCase, base.PersistenceTestMixin):
         conn = self._get_connection()
         conn.clear_all()
         self._backend = None
-        super(MemoryPersistenceTest, self).tearDown()
+        super().tearDown()
 
     def test_memory_backend_entry_point(self):
         conf = {'connection': 'memory:'}

taskflow/tests/unit/persistence/test_sql_persistence.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Rackspace Hosting All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -61,7 +59,7 @@ def _get_connect_string(backend, user, passwd, database=None, variant=None):
         raise Exception("Unrecognized backend: '%s'" % backend)
     if not database:
         database = ''
-    return "%s://%s:%s@localhost/%s" % (backend, user, passwd, database)
+    return "{}://{}:{}@localhost/{}".format(backend, user, passwd, database)
 
 
 def _mysql_exists():
@@ -108,7 +106,7 @@ class SqlitePersistenceTest(test.TestCase, base.PersistenceTestMixin):
         return impl_sqlalchemy.SQLAlchemyBackend(conf).get_connection()
 
     def setUp(self):
-        super(SqlitePersistenceTest, self).setUp()
+        super().setUp()
         self.db_location = tempfile.mktemp(suffix='.db')
         self.db_uri = "sqlite:///%s" % (self.db_location)
         # Ensure upgraded to the right schema
@@ -116,7 +114,7 @@ class SqlitePersistenceTest(test.TestCase, base.PersistenceTestMixin):
             conn.upgrade()
 
     def tearDown(self):
-        super(SqlitePersistenceTest, self).tearDown()
+        super().tearDown()
         if self.db_location and os.path.isfile(self.db_location):
             os.unlink(self.db_location)
             self.db_location = None
@@ -146,7 +144,7 @@ class BackendPersistenceTestMixin(base.PersistenceTestMixin,
         """Cleans up by removing the database once the tests are done."""
 
     def setUp(self):
-        super(BackendPersistenceTestMixin, self).setUp()
+        super().setUp()
         self.backend = None
         try:
             self.db_uri = self._init_db()

taskflow/tests/unit/persistence/test_zk_persistence.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2014 AT&T Labs All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -56,7 +54,7 @@ class ZkPersistenceTest(test.TestCase, base.PersistenceTestMixin):
         return self.backend.get_connection()
 
     def setUp(self):
-        super(ZkPersistenceTest, self).setUp()
+        super().setUp()
         conf = test_utils.ZK_TEST_CONFIG.copy()
         # Create a unique path just for this test (so that we don't overwrite
         # what other tests are doing).
@@ -84,7 +82,7 @@ class ZakePersistenceTest(test.TestCase, base.PersistenceTestMixin):
         return self._backend.get_connection()
 
     def setUp(self):
-        super(ZakePersistenceTest, self).setUp()
+        super().setUp()
         conf = {
             "path": "/taskflow",
         }

taskflow/tests/unit/test_arguments_passing.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -23,11 +21,6 @@ from taskflow import test
 from taskflow.tests import utils
 from taskflow.utils import eventlet_utils as eu
 
-try:
-    from taskflow.engines.action_engine import process_executor as pe
-except ImportError:
-    pe = None
-
 
 class ArgumentsPassingTest(utils.EngineTestBase):
 
@@ -55,9 +48,9 @@ class ArgumentsPassingTest(utils.EngineTestBase):
         }, engine.storage.fetch_all())
 
     def test_save_dict(self):
-        flow = utils.TaskMultiDict(provides=set(['badger',
-                                                 'mushroom',
-                                                 'snake']))
+        flow = utils.TaskMultiDict(provides={'badger',
+                                             'mushroom',
+                                             'snake'})
         engine = self._make_engine(flow)
         engine.run()
         self.assertEqual({
@@ -224,18 +217,3 @@ class ParallelEngineWithEventletTest(ArgumentsPassingTest, test.TestCase):
                                      backend=self.backend,
                                      engine='parallel',
                                      executor=executor)
-
-
-@testtools.skipIf(pe is None, 'process_executor is not available')
-class ParallelEngineWithProcessTest(ArgumentsPassingTest, test.TestCase):
-    _EXECUTOR_WORKERS = 2
-
-    def _make_engine(self, flow, flow_detail=None, executor=None):
-        if executor is None:
-            executor = 'processes'
-        return taskflow.engines.load(flow,
-                                     flow_detail=flow_detail,
-                                     backend=self.backend,
-                                     engine='parallel',
-                                     executor=executor,
-                                     max_workers=self._EXECUTOR_WORKERS)

taskflow/tests/unit/test_check_transition.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -50,7 +48,7 @@ class TransitionTest(test.TestCase):
 class CheckFlowTransitionTest(TransitionTest):
 
     def setUp(self):
-        super(CheckFlowTransitionTest, self).setUp()
+        super().setUp()
         self.check_transition = states.check_flow_transition
         self.transition_exc_regexp = '^Flow transition.*not allowed'
 
@@ -73,7 +71,7 @@ class CheckFlowTransitionTest(TransitionTest):
 class CheckTaskTransitionTest(TransitionTest):
 
     def setUp(self):
-        super(CheckTaskTransitionTest, self).setUp()
+        super().setUp()
         self.check_transition = states.check_task_transition
         self.transition_exc_regexp = '^Task transition.*not allowed'
 
@@ -122,7 +120,7 @@ class CheckTaskTransitionTest(TransitionTest):
 class CheckRetryTransitionTest(CheckTaskTransitionTest):
 
     def setUp(self):
-        super(CheckRetryTransitionTest, self).setUp()
+        super().setUp()
         self.check_transition = states.check_retry_transition
         self.transition_exc_regexp = '^Retry transition.*not allowed'
 

taskflow/tests/unit/test_conductors.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2014 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may

taskflow/tests/unit/test_deciders.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2015 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
@@ -55,4 +53,4 @@ class TestDeciders(test.TestCase):
     def test_bad_pick_widest(self):
         self.assertRaises(ValueError, deciders.pick_widest, [])
         self.assertRaises(ValueError, deciders.pick_widest, ["a"])
-        self.assertRaises(ValueError, deciders.pick_widest, set(['b']))
+        self.assertRaises(ValueError, deciders.pick_widest, {'b'})

taskflow/tests/unit/test_engine_helpers.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) 2013 Yahoo! Inc. All Rights Reserved.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may