fix format

barakalon · Aug 28, 2024 · cd22920 · cd22920
1 parent 89435d8
commit cd22920
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_query.py b/tests/test_query.py
@@ -65,7 +65,9 @@ async def q4(sql: str) -> Sequence[Dict[str, Any]]:
             async with sqlalchemy_engine.connect() as conn:
                 # Sqlglot by-default runs `SET NAMES 'utf8mb4'` if no charset specified, which removes COLLATE settings
                 # See https://github.com/sqlalchemy/sqlalchemy/discussions/7858
-                await conn.execute(text("SET NAMES 'utf8mb4' COLLATE 'utf8mb4_0900_ai_ci'"))
+                await conn.execute(
+                    text("SET NAMES 'utf8mb4' COLLATE 'utf8mb4_0900_ai_ci'")
+                )
                 cursor = await conn.execute(text(sql))
                 if cursor.returns_rows:
                     return cursor.mappings().all()  # type: ignore
@@ -844,7 +846,8 @@ async def test_commands(
         ("KILL 'abc'", "Invalid KILL connection ID"),
         (
             # pick a dynamic string session var
-            "SET init_connect='abc' in xyz", "Unexpected expression: 'abc' IN xyz"
+            "SET init_connect='abc' in xyz",
+            "Unexpected expression: 'abc' IN xyz",
         ),
     ],
 )