fmt

tests/test_jupyter_webapp.py

@@ -47,7 +47,9 @@ class JupyterWebAppTests(unittest.IsolatedAsyncioTestCase):
             response.headers["content-type"].startswith("application/x-ndjson")
         )
         events = [json.loads(line) for line in response.text.splitlines()]
-        result = next(event for event in events if event["msg_type"] == "execute_result")
+        result = next(
+            event for event in events if event["msg_type"] == "execute_result"
+        )
         self.assertEqual(result["content"]["data"]["text/plain"], "{'answer': 42}")
 
     async def test_input_request_can_be_replied_to_while_run_streams(self):
@@ -67,7 +69,9 @@ class JupyterWebAppTests(unittest.IsolatedAsyncioTestCase):
         finally:
             await shell_app.shutdown()
 
-        result = next(event for event in events if event.get("msg_type") == "execute_result")
+        result = next(
+            event for event in events if event.get("msg_type") == "execute_result"
+        )
         self.assertEqual(result["content"]["data"]["text/plain"], "'Ada'")
 
 

tests/test_transport.py

@@ -1,7 +1,11 @@
 import asyncio
 import unittest
 
-from ipython_shell.jupyter.transport import InputRequest, JupyterTransport, KernelMessage
+from ipython_shell.jupyter.transport import (
+    InputRequest,
+    JupyterTransport,
+    KernelMessage,
+)
 
 
 class TransportTests(unittest.TestCase):
@@ -24,9 +28,7 @@ class AsyncTransportTests(unittest.IsolatedAsyncioTestCase):
         await transport.start()
         try:
             call_id = await transport.execute("2 + 2")
-            messages = [
-                message async for message in transport.messages_for(call_id)
-            ]
+            messages = [message async for message in transport.messages_for(call_id)]
         finally:
             await transport.shutdown()
 
@@ -72,19 +74,14 @@ 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)
-            ]
+            messages = [message async for message in transport.messages_for(call_id)]
         finally:
             await transport.shutdown()
 
         display = next(
-            message
-            for message in messages
-            if message.msg_type == "display_data"
+            message for message in messages if message.msg_type == "display_data"
         )
         self.assertIn("image/png", display.content["data"])