fmt, fix

tests/test_info.py

@@ -2,10 +2,10 @@ import unittest
 
 from fastapi.testclient import TestClient
 
+from ipython_agent.ipython_wrapper import get_app_info
 from ipython_mcp.app import app as mcp_app
 from ipython_shell import App, AppInfo
 from ipython_webapp.app import app as web_app
-from ipython_agent.ipython_wrapper import get_app_info
 
 
 class AppInfoTests(unittest.IsolatedAsyncioTestCase):

tests/test_jupyter_app.py

@@ -3,8 +3,8 @@ import re
 import unittest
 
 from ipython_shell.jupyter.app import JupyterApp
-from ipython_shell.jupyter.shell import JupyterShell
 from ipython_shell.jupyter.messages import ExecuteResultMessage
+from ipython_shell.jupyter.shell import JupyterShell
 from ipython_shell.jupyter.transport import InputRequest
 
 

tests/test_transport.py

@@ -211,12 +211,10 @@ class AsyncTransportTests(unittest.IsolatedAsyncioTestCase):
             await transport.start()
             try:
                 call_id = await transport.execute(
-                    "import matplotlib.pyplot as plt\n"
-                    "plt.plot([1, 2, 3], [4, 5, 6])"
+                    "import matplotlib.pyplot as plt\nplt.plot([1, 2, 3], [4, 5, 6])"
                 )
                 messages = [
-                    message
-                    async for message in transport.messages_for(call_id)
+                    message async for message in transport.messages_for(call_id)
                 ]
             finally:
                 await transport.shutdown()