refactor: move imports to top of integration test files

- Move firebolt.db and firebolt.async_db imports to top of test files
- Remove local import statements from inside test functions
- Use full module names instead of local aliases

Addresses final PR feedback from ptiurin on #463

Co-Authored-By: petro.tiurin@firebolt.io <petro.tiurin@firebolt.io>

Author: devin-ai-integration[bot]

tests/integration/dbapi/async/V2/test_queries_async.py

@@ -6,6 +6,7 @@
 
 from pytest import mark, raises
 
+import firebolt.async_db
 from firebolt.async_db import Binary, Connection, Cursor, OperationalError
 from firebolt.async_db.connection import connect
 from firebolt.client.auth.base import Auth
@@ -285,12 +286,10 @@ async def test_parameterized_query_with_special_chars(connection: Connection) ->
 @mark.parametrize("paramstyle", ["qmark", "fb_numeric"])
 async def test_executemany_bulk_insert(connection: Connection, paramstyle: str) -> None:
     """executemany with bulk_insert=True inserts data correctly."""
-    import firebolt.async_db as db_module
-
-    original_paramstyle = db_module.paramstyle
+    original_paramstyle = firebolt.async_db.paramstyle
 
     try:
-        db_module.paramstyle = paramstyle
+        firebolt.async_db.paramstyle = paramstyle
 
         async with connection.cursor() as c:
             await c.execute('DROP TABLE IF EXISTS "test_bulk_insert_async"')
@@ -320,7 +319,7 @@ async def test_executemany_bulk_insert(connection: Connection, paramstyle: str)
 
             await c.execute('DROP TABLE "test_bulk_insert_async"')
     finally:
-        db_module.paramstyle = original_paramstyle
+        firebolt.async_db.paramstyle = original_paramstyle
 
 
 async def test_multi_statement_query(connection: Connection) -> None:

tests/integration/dbapi/sync/V2/test_queries.py

@@ -7,6 +7,7 @@
 
 from pytest import mark, raises
 
+import firebolt.db
 from firebolt.client.auth import Auth
 from firebolt.common._types import ColType
 from firebolt.common.row_set.types import Column
@@ -286,12 +287,10 @@ def test_empty_query(c: Cursor, query: str, params: tuple) -> None:
 @mark.parametrize("paramstyle", ["qmark", "fb_numeric"])
 def test_executemany_bulk_insert(connection: Connection, paramstyle: str) -> None:
     """executemany with bulk_insert=True inserts data correctly."""
-    import firebolt.db as db_module
-
-    original_paramstyle = db_module.paramstyle
+    original_paramstyle = firebolt.db.paramstyle
 
     try:
-        db_module.paramstyle = paramstyle
+        firebolt.db.paramstyle = paramstyle
 
         with connection.cursor() as c:
             c.execute('DROP TABLE IF EXISTS "test_bulk_insert"')
@@ -321,7 +320,7 @@ def test_executemany_bulk_insert(connection: Connection, paramstyle: str) -> Non
 
             c.execute('DROP TABLE "test_bulk_insert"')
     finally:
-        db_module.paramstyle = original_paramstyle
+        firebolt.db.paramstyle = original_paramstyle
 
 
 def test_multi_statement_query(connection: Connection) -> None: