make lint happy

DiamondLightSource · Mar 11, 2024 · 1ebf6a9 · 1ebf6a9
1 parent bb491d1
commit 1ebf6a9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 18 deletions.
diff --git 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
@@ -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() == []