⚫

tests/engine/test_task_runner.py

@@ -830,8 +830,8 @@ def test_task_handler_that_doesnt_return_state(self):
             with prefect.utilities.tests.raise_on_exception():
                 TaskRunner(task=task).run()
 
-class TestTaskRunnerStateHandlers:
 
+class TestTaskRunnerStateHandlers:
     def test_task_runner_handlers_are_called(self):
         TaskRunner(task=Task(), state_handlers=[task_runner_handler]).run()
         # the task changed state twice: Pending -> Running -> Success
@@ -868,11 +868,12 @@ def test_task_runner_handler_that_doesnt_return_state(self):
         # doesn't exist on None
         with pytest.raises(AttributeError):
             with prefect.utilities.tests.raise_on_exception():
-                TaskRunner(task=Task(), state_handlers=[lambda *a:None]).run()
+                TaskRunner(task=Task(), state_handlers=[lambda *a: None]).run()
 
     def test_task_handler_that_raises_signal_is_trapped(self):
         def handler(task, old, new):
             raise signals.Fail()
+
         task = Task(state_handlers=[handler])
         state = TaskRunner(task=task).run()
         assert state.is_failed()