diff options
Diffstat (limited to 'taskflow/tests/unit/test_engines.py')
-rw-r--r-- | taskflow/tests/unit/test_engines.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/taskflow/tests/unit/test_engines.py b/taskflow/tests/unit/test_engines.py index 81f0e08..4454429 100644 --- a/taskflow/tests/unit/test_engines.py +++ b/taskflow/tests/unit/test_engines.py @@ -20,7 +20,6 @@ import functools import threading import futurist -import six import testtools import taskflow.engines @@ -350,7 +349,7 @@ class EngineLinearFlowTest(utils.EngineTestBase): engine_it = engine.run_iter() while True: try: - engine_state = six.next(engine_it) + engine_state = next(engine_it) if engine_state not in engine_states: engine_states[engine_state] = 1 else: @@ -1318,7 +1317,7 @@ class EngineGraphConditionalFlowTest(utils.EngineTestBase): def test_graph_flow_conditional_history(self): def even_odd_decider(history, allowed): - total = sum(six.itervalues(history)) + total = sum(history.values()) if total == allowed: return True return False |