style: clean up imports and comments in test files

- Revert unrelated import order and whitespace changes in integration __init__.py
- Fix Db2 capitalization in test_dialect.py comment

Signed-off-by: IBM Db2 Eco System <Hdm-dev-persona-db2-eco-system@ibm.com>

Author: hdm-db2-eco-system

tests/core/engine_adapter/integration/__init__.py

@@ -3,30 +3,30 @@
 import os
 import pathlib
 import sys
-import time
 import typing as t
+import time
 from contextlib import contextmanager
-from dataclasses import dataclass
 
 import pandas as pd  # noqa: TID253
 import pytest
-from _pytest.mark import MarkDecorator
-from _pytest.mark.structures import ParameterSet
 from sqlglot import exp, parse_one
 from sqlglot.optimizer.normalize_identifiers import normalize_identifiers
 
-import sqlmesh.core.dialect as d
 from sqlmesh import Config, Context, EngineAdapter
 from sqlmesh.core.config import load_config_from_paths
 from sqlmesh.core.config.connection import AthenaConnectionConfig
 from sqlmesh.core.dialect import normalize_model_name
-from sqlmesh.core.engine_adapter import AthenaEngineAdapter, SparkEngineAdapter, TrinoEngineAdapter
+import sqlmesh.core.dialect as d
+from sqlmesh.core.engine_adapter import SparkEngineAdapter, TrinoEngineAdapter, AthenaEngineAdapter
 from sqlmesh.core.engine_adapter.shared import DataObject
 from sqlmesh.core.model.definition import SqlModel, load_sql_based_model
 from sqlmesh.utils import random_id
 from sqlmesh.utils.date import to_ds
 from sqlmesh.utils.pydantic import PydanticModel
 from tests.utils.pandas import compare_dataframes
+from dataclasses import dataclass
+from _pytest.mark import MarkDecorator
+from _pytest.mark.structures import ParameterSet
 
 if t.TYPE_CHECKING:
     from sqlmesh.core._typing import TableName, SchemaName
@@ -658,7 +658,7 @@ def get_column_comments(
             query = f"""
                 SELECT
                     col.COLUMN_NAME column_name,
-                    CAST(ep.value AS NVARCHAR(MAX)) comment
+                    CAST(ep.value AS NVARCHAR(MAX)) comment 
                 FROM INFORMATION_SCHEMA.COLUMNS col
                 CROSS APPLY fn_listextendedproperty('MS_Description', 'schema', col.TABLE_SCHEMA, '{kind}', col.TABLE_NAME, 'column', col.COLUMN_NAME) ep
                 WHERE col.TABLE_SCHEMA = '{schema_name}' AND col.TABLE_NAME = '{table_name}'

tests/core/test_dialect.py

@@ -1051,7 +1051,7 @@ def test_parse_snowflake_create_schema_ddl():
 
 @pytest.mark.parametrize("dialect", sorted(set(DIALECT_TO_TYPE.values())))
 def test_sqlglot_extended_correctly(dialect: str) -> None:
-    # Skip DB2 on Python 3.9 since db2-sqlglot-dialect requires Python 3.10+
+    # Skip Db2 on Python 3.9 since db2-sqlglot-dialect requires Python 3.10+
     if dialect == "db2" and sys.version_info < (3, 10):
         pytest.skip("DB2 dialect requires Python 3.10+ for db2-sqlglot-dialect")