chore: format the new assertions at the repo's 70-char line length

black==24.2.0 --check is a CI gate and these two hunks were over it, so
lint was red on this branch. Behaviour unchanged: one call assertion
wrapped, one docstring closer moved to its own line.

Author: ayaangazali

tests/structs/test_reasoning_agent_router.py

@@ -349,7 +349,9 @@ def test_run_forwards_kwargs_and_rejects_stray_positionals():
         router, "select_swarm", return_value=mock_swarm
     ):
         router.run("task", key="value")
-        mock_swarm.run.assert_called_once_with(task="task", key="value")
+        mock_swarm.run.assert_called_once_with(
+            task="task", key="value"
+        )
 
         with pytest.raises(TypeError):
             router.run("task", "extra_arg")
@@ -414,7 +416,8 @@ def test_batched_run_empty_list():
 
 def test_batched_run_forwards_kwargs_to_every_task():
     """batched_run reached self.run(task) with no kwargs, so img,
-    streaming_callback and everything else were dropped for batched callers."""
+    streaming_callback and everything else were dropped for batched callers.
+    """
     router = ReasoningAgentRouter(swarm_type="reasoning-duo")
     with patch.object(router, "run", return_value="x") as mock_run:
         router.batched_run(["a", "b"], kw="val")