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

taskflow/tests/unit/test_storage.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
@@ -30,9 +28,9 @@ from taskflow.types import failure
 from taskflow.utils import persistence_utils as p_utils
 
 
-class StorageTestMixin(object):
+class StorageTestMixin:
     def setUp(self):
-        super(StorageTestMixin, self).setUp()
+        super().setUp()
         self.backend = None
         self.thread_count = 50
 
@@ -40,7 +38,7 @@ class StorageTestMixin(object):
         with contextlib.closing(self.backend) as be:
             with contextlib.closing(be.get_connection()) as conn:
                 conn.clear_all()
-        super(StorageTestMixin, self).tearDown()
+        super().tearDown()
 
     @staticmethod
     def _run_many_threads(threads):
@@ -357,14 +355,14 @@ class StorageTestMixin(object):
         s.inject({'foo': 'bar', 'spam': 'eggs'})
         self.assertEqual({'viking': 'eggs'},
                          s.fetch_mapped_args({'viking': 'spam'},
-                                             optional_args=set(['viking'])))
+                                             optional_args={'viking'}))
 
     def test_fetch_optional_args_not_found(self):
         s = self._get_storage()
         s.inject({'foo': 'bar', 'spam': 'eggs'})
         self.assertEqual({},
                          s.fetch_mapped_args({'viking': 'helmet'},
-                                             optional_args=set(['viking'])))
+                                             optional_args={'viking'}))
 
     def test_set_and_get_task_state(self):
         s = self._get_storage()
@@ -437,7 +435,7 @@ class StorageTestMixin(object):
 
     def test_result_is_checked(self):
         s = self._get_storage()
-        s.ensure_atom(test_utils.NoopTask('my task', provides=set(['result'])))
+        s.ensure_atom(test_utils.NoopTask('my task', provides={'result'}))
         s.save('my task', {})
         self.assertRaisesRegex(exceptions.NotFound,
                                '^Unable to find result', s.fetch, 'result')
@@ -539,7 +537,7 @@ class StorageTestMixin(object):
         s = self._get_storage()
         s.ensure_atom(t)
         missing = s.fetch_unsatisfied_args(t.name, t.rebind)
-        self.assertEqual(set(['x']), missing)
+        self.assertEqual({'x'}, missing)
         s.inject_atom_args(t.name, {'x': 2}, transient=False)
         missing = s.fetch_unsatisfied_args(t.name, t.rebind)
         self.assertEqual(set(), missing)
@@ -551,7 +549,7 @@ class StorageTestMixin(object):
         s = self._get_storage()
         s.ensure_atom(t)
         missing = s.fetch_unsatisfied_args(t.name, t.rebind)
-        self.assertEqual(set(['x']), missing)
+        self.assertEqual({'x'}, missing)
         s.inject_atom_args(t.name, {'x': 2}, transient=False)
         s.inject_atom_args(t.name, {'x': 3}, transient=True)
         missing = s.fetch_unsatisfied_args(t.name, t.rebind)
@@ -589,13 +587,13 @@ class StorageTestMixin(object):
 
 class StorageMemoryTest(StorageTestMixin, test.TestCase):
     def setUp(self):
-        super(StorageMemoryTest, self).setUp()
+        super().setUp()
         self.backend = backends.fetch({'connection': 'memory://'})
 
 
 class StorageSQLTest(StorageTestMixin, test.TestCase):
     def setUp(self):
-        super(StorageSQLTest, self).setUp()
+        super().setUp()
         self.backend = backends.fetch({'connection': 'sqlite://'})
         with contextlib.closing(self.backend.get_connection()) as conn:
             conn.upgrade()

taskflow/tests/unit/test_suspend.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
@@ -25,23 +23,18 @@ 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 SuspendingListener(utils.CaptureListener):
 
     def __init__(self, engine,
                  task_name, task_state, capture_flow=False):
-        super(SuspendingListener, self).__init__(
+        super().__init__(
             engine,
             capture_flow=capture_flow)
         self._revert_match = (task_name, task_state)
 
     def _task_receiver(self, state, details):
-        super(SuspendingListener, self)._task_receiver(state, details)
+        super()._task_receiver(state, details)
         if (details['task_name'], state) == self._revert_match:
             self._engine.suspend()
 
@@ -227,17 +220,3 @@ class ParallelEngineWithEventletTest(SuspendTest, test.TestCase):
         return taskflow.engines.load(flow, flow_detail=flow_detail,
                                      backend=self.backend, engine='parallel',
                                      executor=executor)
-
-
-@testtools.skipIf(pe is None, 'process_executor is not available')
-class ParallelEngineWithProcessTest(SuspendTest, 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,
-                                     engine='parallel',
-                                     backend=self.backend,
-                                     executor=executor,
-                                     max_workers=self._EXECUTOR_WORKERS)

taskflow/tests/unit/test_task.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright 2015 Hewlett-Packard Development Company, L.P.
 #    Copyright (C) 2013 Yahoo! Inc. All Rights Reserved.
 #
@@ -81,7 +79,7 @@ class TaskTest(test.TestCase):
 
     def test_generated_name(self):
         my_task = MyTask()
-        self.assertEqual('%s.%s' % (__name__, 'MyTask'),
+        self.assertEqual('{}.{}'.format(__name__, 'MyTask'),
                          my_task.name)
 
     def test_task_str(self):
@@ -121,7 +119,7 @@ class TaskTest(test.TestCase):
         }
         self.assertEqual(expected,
                          my_task.rebind)
-        self.assertEqual(set(['spam', 'eggs', 'context']),
+        self.assertEqual({'spam', 'eggs', 'context'},
                          my_task.requires)
 
     def test_requires_amended(self):
@@ -150,12 +148,12 @@ class TaskTest(test.TestCase):
 
     def test_requires_ignores_optional(self):
         my_task = DefaultArgTask()
-        self.assertEqual(set(['spam']), my_task.requires)
-        self.assertEqual(set(['eggs']), my_task.optional)
+        self.assertEqual({'spam'}, my_task.requires)
+        self.assertEqual({'eggs'}, my_task.optional)
 
     def test_requires_allows_optional(self):
         my_task = DefaultArgTask(requires=('spam', 'eggs'))
-        self.assertEqual(set(['spam', 'eggs']), my_task.requires)
+        self.assertEqual({'spam', 'eggs'}, my_task.requires)
         self.assertEqual(set(), my_task.optional)
 
     def test_rebind_includes_optional(self):
@@ -174,7 +172,7 @@ class TaskTest(test.TestCase):
             'context': 'c'
         }
         self.assertEqual(expected, my_task.rebind)
-        self.assertEqual(set(['a', 'b', 'c']),
+        self.assertEqual({'a', 'b', 'c'},
                          my_task.requires)
 
     def test_rebind_partial(self):
@@ -185,7 +183,7 @@ class TaskTest(test.TestCase):
             'context': 'context'
         }
         self.assertEqual(expected, my_task.rebind)
-        self.assertEqual(set(['a', 'b', 'context']),
+        self.assertEqual({'a', 'b', 'context'},
                          my_task.requires)
 
     def test_rebind_unknown(self):
@@ -208,7 +206,7 @@ class TaskTest(test.TestCase):
             'eggs': 'c'
         }
         self.assertEqual(expected, my_task.rebind)
-        self.assertEqual(set(['a', 'b', 'c']),
+        self.assertEqual({'a', 'b', 'c'},
                          my_task.requires)
 
     def test_rebind_list_partial(self):
@@ -219,7 +217,7 @@ class TaskTest(test.TestCase):
             'eggs': 'eggs'
         }
         self.assertEqual(expected, my_task.rebind)
-        self.assertEqual(set(['a', 'b', 'eggs']),
+        self.assertEqual({'a', 'b', 'eggs'},
                          my_task.requires)
 
     def test_rebind_list_more(self):
@@ -234,7 +232,7 @@ class TaskTest(test.TestCase):
             'c': 'c'
         }
         self.assertEqual(expected, my_task.rebind)
-        self.assertEqual(set(['a', 'b', 'c']),
+        self.assertEqual({'a', 'b', 'c'},
                          my_task.requires)
 
     def test_rebind_list_bad_value(self):
@@ -243,12 +241,12 @@ class TaskTest(test.TestCase):
 
     def test_default_provides(self):
         my_task = DefaultProvidesTask()
-        self.assertEqual(set(['def']), my_task.provides)
+        self.assertEqual({'def'}, my_task.provides)
         self.assertEqual({'def': None}, my_task.save_as)
 
     def test_default_provides_can_be_overridden(self):
         my_task = DefaultProvidesTask(provides=('spam', 'eggs'))
-        self.assertEqual(set(['spam', 'eggs']), my_task.provides)
+        self.assertEqual({'spam', 'eggs'}, my_task.provides)
         self.assertEqual({'spam': 0, 'eggs': 1}, my_task.save_as)
 
     def test_update_progress_within_bounds(self):
@@ -366,7 +364,7 @@ class TaskTest(test.TestCase):
         my_task = SeparateRevertTask(rebind=('a',), revert_rebind=('b',))
         self.assertEqual({'execute_arg': 'a'}, my_task.rebind)
         self.assertEqual({'revert_arg': 'b'}, my_task.revert_rebind)
-        self.assertEqual(set(['a', 'b']),
+        self.assertEqual({'a', 'b'},
                          my_task.requires)
 
         my_task = SeparateRevertTask(requires='execute_arg',
@@ -374,13 +372,13 @@ class TaskTest(test.TestCase):
 
         self.assertEqual({'execute_arg': 'execute_arg'}, my_task.rebind)
         self.assertEqual({'revert_arg': 'revert_arg'}, my_task.revert_rebind)
-        self.assertEqual(set(['execute_arg', 'revert_arg']),
+        self.assertEqual({'execute_arg', 'revert_arg'},
                          my_task.requires)
 
     def test_separate_revert_optional_args(self):
         my_task = SeparateRevertOptionalTask()
-        self.assertEqual(set(['execute_arg']), my_task.optional)
-        self.assertEqual(set(['revert_arg']), my_task.revert_optional)
+        self.assertEqual({'execute_arg'}, my_task.optional)
+        self.assertEqual({'revert_arg'}, my_task.revert_optional)
 
     def test_revert_kwargs(self):
         my_task = RevertKwargsTask()
@@ -389,7 +387,7 @@ class TaskTest(test.TestCase):
         self.assertEqual(expected_rebind, my_task.rebind)
         expected_rebind = {'execute_arg1': 'execute_arg1'}
         self.assertEqual(expected_rebind, my_task.revert_rebind)
-        self.assertEqual(set(['execute_arg1', 'execute_arg2']),
+        self.assertEqual({'execute_arg1', 'execute_arg2'},
                          my_task.requires)
 
 

taskflow/tests/unit/test_types.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
@@ -66,9 +64,9 @@ class GraphTest(test.TestCase):
         g.add_node("b")
         g.add_node("c")
         g.add_edge("b", "c")
-        self.assertEqual(set(['a', 'b']),
+        self.assertEqual({'a', 'b'},
                          set(g.no_predecessors_iter()))
-        self.assertEqual(set(['a', 'c']),
+        self.assertEqual({'a', 'c'},
                          set(g.no_successors_iter()))
 
     def test_directed(self):
@@ -534,8 +532,8 @@ CEO
     def test_dfs_itr(self):
         root = self._make_species()
         things = list([n.item for n in root.dfs_iter(include_self=True)])
-        self.assertEqual(set(['animal', 'reptile', 'mammal', 'horse',
-                              'primate', 'monkey', 'human']), set(things))
+        self.assertEqual({'animal', 'reptile', 'mammal', 'horse',
+                          'primate', 'monkey', 'human'}, set(things))
 
     def test_dfs_itr_left_to_right(self):
         root = self._make_species()

taskflow/tests/unit/test_utils.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
@@ -30,7 +28,7 @@ from taskflow.utils import threading_utils
 class CachedPropertyTest(test.TestCase):
     def test_attribute_caching(self):
 
-        class A(object):
+        class A:
             def __init__(self):
                 self.call_counter = 0
 
@@ -46,7 +44,7 @@ class CachedPropertyTest(test.TestCase):
 
     def test_custom_property(self):
 
-        class A(object):
+        class A:
             @misc.cachedproperty('_c')
             def b(self):
                 return 'b'
@@ -60,7 +58,7 @@ class CachedPropertyTest(test.TestCase):
         def try_del(a):
             del a.b
 
-        class A(object):
+        class A:
             @misc.cachedproperty
             def b(self):
                 return 'b'
@@ -75,7 +73,7 @@ class CachedPropertyTest(test.TestCase):
         def try_set(a):
             a.b = 'c'
 
-        class A(object):
+        class A:
             @misc.cachedproperty
             def b(self):
                 return 'b'
@@ -87,7 +85,7 @@ class CachedPropertyTest(test.TestCase):
 
     def test_documented_property(self):
 
-        class A(object):
+        class A:
             @misc.cachedproperty
             def b(self):
                 """I like bees."""
@@ -97,7 +95,7 @@ class CachedPropertyTest(test.TestCase):
 
     def test_undocumented_property(self):
 
-        class A(object):
+        class A:
             @misc.cachedproperty
             def b(self):
                 return 'b'
@@ -107,7 +105,7 @@ class CachedPropertyTest(test.TestCase):
     def test_threaded_access_property(self):
         called = collections.deque()
 
-        class A(object):
+        class A:
             @misc.cachedproperty
             def b(self):
                 called.append(1)
@@ -311,7 +309,7 @@ class TestClamping(test.TestCase):
 class TestIterable(test.TestCase):
     def test_string_types(self):
         self.assertFalse(misc.is_iterable('string'))
-        self.assertFalse(misc.is_iterable(u'string'))
+        self.assertFalse(misc.is_iterable('string'))
 
     def test_list(self):
         self.assertTrue(misc.is_iterable(list()))
@@ -347,7 +345,7 @@ class TestSafeCopyDictRaises(testscenarios.TestWithScenarios):
     scenarios = [
         ('list', {'original': [1, 2], 'exception': TypeError}),
         ('tuple', {'original': (1, 2), 'exception': TypeError}),
-        ('set', {'original': set([1, 2]), 'exception': TypeError}),
+        ('set', {'original': {1, 2}, 'exception': TypeError}),
     ]
 
     def test_exceptions(self):

taskflow/tests/unit/test_utils_async_utils.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

taskflow/tests/unit/test_utils_binary.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
@@ -38,15 +36,15 @@ class BinaryEncodeTest(test.TestCase):
         self._check(data, data)
 
     def test_simple_text(self):
-        self._check(u'hello', _bytes('hello'))
+        self._check('hello', _bytes('hello'))
 
     def test_unicode_text(self):
-        self._check(u'привет', _bytes('привет'))
+        self._check('привет', _bytes('привет'))
 
     def test_unicode_other_encoding(self):
-        result = misc.binary_encode(u'mañana', 'latin-1')
+        result = misc.binary_encode('mañana', 'latin-1')
         self.assertIsInstance(result, bytes)
-        self.assertEqual(u'mañana'.encode('latin-1'), result)
+        self.assertEqual('mañana'.encode('latin-1'), result)
 
 
 class BinaryDecodeTest(test.TestCase):
@@ -57,24 +55,24 @@ class BinaryDecodeTest(test.TestCase):
         self.assertEqual(expected_result, result)
 
     def test_simple_text(self):
-        data = u'hello'
+        data = 'hello'
         self._check(data, data)
 
     def test_unicode_text(self):
-        data = u'привет'
+        data = 'привет'
         self._check(data, data)
 
     def test_simple_binary(self):
-        self._check(_bytes('hello'), u'hello')
+        self._check(_bytes('hello'), 'hello')
 
     def test_unicode_binary(self):
-        self._check(_bytes('привет'), u'привет')
+        self._check(_bytes('привет'), 'привет')
 
     def test_unicode_other_encoding(self):
-        data = u'mañana'.encode('latin-1')
+        data = 'mañana'.encode('latin-1')
         result = misc.binary_decode(data, 'latin-1')
         self.assertIsInstance(result, str)
-        self.assertEqual(u'mañana', result)
+        self.assertEqual('mañana', result)
 
 
 class DecodeJsonTest(test.TestCase):
@@ -85,11 +83,11 @@ class DecodeJsonTest(test.TestCase):
 
     def test_it_works_with_unicode(self):
         data = _bytes('{"foo": "фуу"}')
-        self.assertEqual({"foo": u'фуу'}, misc.decode_json(data))
+        self.assertEqual({"foo": 'фуу'}, misc.decode_json(data))
 
     def test_handles_invalid_unicode(self):
         self.assertRaises(ValueError, misc.decode_json,
-                          '{"\xf1": 1}'.encode('latin-1'))
+                          b'{"\xf1": 1}')
 
     def test_handles_bad_json(self):
         self.assertRaises(ValueError, misc.decode_json,

taskflow/tests/unit/test_utils_iter_utils.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
@@ -140,7 +138,7 @@ class IterUtilsTest(test.TestCase):
         self.assertRaises(ValueError, iter_utils.while_is_not, 2, 'a')
 
     def test_while_is_not(self):
-        class Dummy(object):
+        class Dummy:
             def __init__(self, char):
                 self.char = char
         dummy_list = [Dummy(a)

taskflow/tests/unit/test_utils_kazoo_utils.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 #    Copyright (C) Red Hat
 #
 #    Licensed under the Apache License, Version 2.0 (the "License"); you may

taskflow/tests/unit/test_utils_threading_utils.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
@@ -53,7 +51,7 @@ class TestThreadBundle(test.TestCase):
     thread_count = 5
 
     def setUp(self):
-        super(TestThreadBundle, self).setUp()
+        super().setUp()
         self.bundle = tu.ThreadBundle()
         self.death = threading.Event()
         self.addCleanup(self.bundle.stop)

taskflow/tests/unit/worker_based/test_creation.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/worker_based/test_dispatcher.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/worker_based/test_endpoint.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
@@ -25,7 +23,7 @@ from taskflow.tests import utils
 class Task(task.Task):
 
     def __init__(self, a, *args, **kwargs):
-        super(Task, self).__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
 
     def execute(self, *args, **kwargs):
         pass
@@ -34,7 +32,7 @@ class Task(task.Task):
 class TestEndpoint(test.TestCase):
 
     def setUp(self):
-        super(TestEndpoint, self).setUp()
+        super().setUp()
         self.task_cls = utils.TaskOneReturn
         self.task_uuid = 'task-uuid'
         self.task_args = {'context': 'context'}

taskflow/tests/unit/worker_based/test_executor.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
@@ -29,7 +27,7 @@ from taskflow.types import failure
 class TestWorkerTaskExecutor(test.MockTestCase):
 
     def setUp(self):
-        super(TestWorkerTaskExecutor, self).setUp()
+        super().setUp()
         self.task = test_utils.DummyTask()
         self.task_uuid = 'task-uuid'
         self.task_args = {'a': 'a'}

taskflow/tests/unit/worker_based/test_message_pump.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/worker_based/test_pipeline.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/worker_based/test_protocol.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
@@ -25,7 +23,7 @@ from taskflow.tests import utils
 from taskflow.types import failure
 
 
-class Unserializable(object):
+class Unserializable:
     pass
 
 
@@ -98,7 +96,7 @@ class TestProtocolValidation(test.TestCase):
 class TestProtocol(test.TestCase):
 
     def setUp(self):
-        super(TestProtocol, self).setUp()
+        super().setUp()
         self.task = utils.DummyTask()
         self.task_uuid = 'task-uuid'
         self.task_action = 'execute'

taskflow/tests/unit/worker_based/test_proxy.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
@@ -25,7 +23,7 @@ from taskflow.utils import threading_utils
 class TestProxy(test.MockTestCase):
 
     def setUp(self):
-        super(TestProxy, self).setUp()
+        super().setUp()
         self.topic = 'test-topic'
         self.broker_url = 'test-url'
         self.exchange = 'test-exchange'
@@ -72,7 +70,7 @@ class TestProxy(test.MockTestCase):
         self.resetMasterMock()
 
     def _queue_name(self, topic):
-        return "%s_%s" % (self.exchange, topic)
+        return "{}_{}".format(self.exchange, topic)
 
     def proxy_start_calls(self, calls, exc_type=mock.ANY):
         return [

taskflow/tests/unit/worker_based/test_server.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
@@ -27,7 +25,7 @@ from taskflow.types import failure
 class TestServer(test.MockTestCase):
 
     def setUp(self):
-        super(TestServer, self).setUp()
+        super().setUp()
         self.server_topic = 'server-topic'
         self.server_exchange = 'server-exchange'
         self.broker_url = 'test-url'
@@ -140,8 +138,8 @@ class TestServer(test.MockTestCase):
         self.assertEqual(
             (self.task.name, self.task.name, 'revert',
              dict(arguments=self.task_args,
-                  failures=dict((i, utils.FailureMatcher(f))
-                                for i, f in failures.items()))),
+                  failures={i: utils.FailureMatcher(f)
+                            for i, f in failures.items()})),
             (task_cls, task_name, action, task_args))
 
     @mock.patch("taskflow.engines.worker_based.server.LOG.critical")

taskflow/tests/unit/worker_based/test_types.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/worker_based/test_worker.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
@@ -27,7 +25,7 @@ from taskflow.tests import utils
 class TestWorker(test.MockTestCase):
 
     def setUp(self):
-        super(TestWorker, self).setUp()
+        super().setUp()
         self.task_cls = utils.DummyTask
         self.task_name = reflection.get_class_name(self.task_cls)
         self.broker_url = 'test-url'