diff --git a/tests/service/test_subprocess_handler.py b/tests/service/test_subprocess_handler.py index 556c56dd4..aa64e0488 100644 --- a/tests/service/test_subprocess_handler.py +++ b/tests/service/test_subprocess_handler.py @@ -138,9 +138,7 @@ def run_in_same_process(func, args=None): Task(name="sleep", params={"time": 0.0}) ) == dummy_handler.submit_task(Task(name="sleep", params={"time": 0.0})) - assert sp_handler.clear_task("task_id") == dummy_handler.clear_task( - "task_id" - ) + assert sp_handler.clear_task("task_id") == dummy_handler.clear_task("task_id") assert sp_handler.begin_task(WorkerTask(task_id="foo")) == dummy_handler.begin_task( WorkerTask(task_id="foo") diff --git a/tests/worker/test_reworker.py b/tests/worker/test_reworker.py index 667f7e031..d914f3864 100644 --- a/tests/worker/test_reworker.py +++ b/tests/worker/test_reworker.py @@ -110,17 +110,13 @@ def test_multi_start(inert_worker: Worker) -> None: def test_submit_task(worker: Worker) -> None: assert worker.get_tasks() == [] task_id = worker.submit_task(_SIMPLE_TASK) - assert worker.get_tasks() == [ - TrackableTask(task_id=task_id, task=_SIMPLE_TASK) - ] + assert worker.get_tasks() == [TrackableTask(task_id=task_id, task=_SIMPLE_TASK)] def test_submit_multiple_tasks(worker: Worker) -> None: assert worker.get_tasks() == [] task_id_1 = worker.submit_task(_SIMPLE_TASK) - assert worker.get_tasks() == [ - TrackableTask(task_id=task_id_1, task=_SIMPLE_TASK) - ] + assert worker.get_tasks() == [TrackableTask(task_id=task_id_1, task=_SIMPLE_TASK)] task_id_2 = worker.submit_task(_LONG_TASK) assert worker.get_tasks() == [ TrackableTask(task_id=task_id_1, task=_SIMPLE_TASK), @@ -136,14 +132,10 @@ def test_stop_with_task_pending(inert_worker: Worker) -> None: def test_restart_leaves_task_pending(worker: Worker) -> None: task_id = worker.submit_task(_SIMPLE_TASK) - assert worker.get_tasks() == [ - TrackableTask(task_id=task_id, task=_SIMPLE_TASK) - ] + assert worker.get_tasks() == [TrackableTask(task_id=task_id, task=_SIMPLE_TASK)] worker.stop() worker.start() - assert worker.get_tasks() == [ - TrackableTask(task_id=task_id, task=_SIMPLE_TASK) - ] + assert worker.get_tasks() == [TrackableTask(task_id=task_id, task=_SIMPLE_TASK)] def test_submit_before_start_pending(inert_worker: Worker) -> None: @@ -160,9 +152,7 @@ def test_submit_before_start_pending(inert_worker: Worker) -> None: def test_clear_task(worker: Worker) -> None: task_id = worker.submit_task(_SIMPLE_TASK) - assert worker.get_tasks() == [ - TrackableTask(task_id=task_id, task=_SIMPLE_TASK) - ] + assert worker.get_tasks() == [TrackableTask(task_id=task_id, task=_SIMPLE_TASK)] assert worker.clear_task(task_id) assert worker.get_tasks() == []