diff --git a/opentech/apply/tests/test_workflow.py b/opentech/apply/tests/test_workflow.py index cabd5a0df52c4427ce90aee3095da678701e7e8a..1fda36a3184b2665b960ef115fafc2bfe66409eb 100644 --- a/opentech/apply/tests/test_workflow.py +++ b/opentech/apply/tests/test_workflow.py @@ -32,6 +32,7 @@ class TestWorkflowCreation(SimpleTestCase): workflow = WorkflowFactory(num_stages=2) self.assertEqual(workflow.next(workflow.stages[0]), workflow.stages[1]) + class TestStageCreation(SimpleTestCase): def test_can_create_stage(self): name = 'the_stage' diff --git a/opentech/apply/workflow.py b/opentech/apply/workflow.py index 7232ef4472e9a16bcfc4878ebedfa7ac35ed53ca..60d4fedf4e42b435af1a3997a3f79c3643f22b8a 100644 --- a/opentech/apply/workflow.py +++ b/opentech/apply/workflow.py @@ -18,7 +18,7 @@ class Workflow(Iterable['Stage']): for i, stage in enumerate(self): if stage == current_stage: try: - return self.stages[i+1] + return self.stages[i + 1] except IndexError: pass