Linting fix for CI to pass

Author: sfc-gh-fpawlowski

src/snowflake/connector/aio/_pandas_tools.py

@@ -8,6 +8,13 @@
 
 from snowflake.connector import ProgrammingError
 from snowflake.connector.options import pandas
+
+# Import utilities from sync version
+from snowflake.connector.pandas_tools import (
+    _iceberg_config_statement_helper,
+    build_location_helper,
+    chunk_helper,
+)
 from snowflake.connector.telemetry import TelemetryData, TelemetryField
 
 from .._utils import (
@@ -16,13 +23,6 @@
     random_name_for_temp_object,
 )
 from ..constants import _PARAM_USE_SCOPED_TEMP_FOR_PANDAS_TOOLS
-
-# Import utilities from sync version
-from ..pandas_tools import (
-    _iceberg_config_statement_helper,
-    build_location_helper,
-    chunk_helper,
-)
 from ._cursor import SnowflakeCursor
 
 if TYPE_CHECKING:  # pragma: no cover

test/integ/aio_it/test_pandas_tools_async.py

@@ -2,7 +2,6 @@
 from __future__ import annotations
 
 import math
-import re
 from datetime import datetime, timedelta, timezone
 from typing import TYPE_CHECKING, Any, Callable, Generator
 from unittest import mock
@@ -24,11 +23,9 @@
 try:
     from snowflake.connector.aio._pandas_tools import write_pandas
     from snowflake.connector.options import pandas
-    from snowflake.connector.pandas_tools import _iceberg_config_statement_helper
 except ImportError:
     pandas = None
     write_pandas = None
-    _iceberg_config_statement_helper = None
 
 if TYPE_CHECKING:
     from snowflake.connector.aio import SnowflakeConnection
@@ -1053,34 +1050,6 @@ async def mock_execute(*args, **kwargs):
             await cnx.execute_string(f"drop schema if exists {target_schema}")
 
 
-def test__iceberg_config_statement_helper():
-    config = {
-        "EXTERNAL_VOLUME": "vol",
-        "CATALOG": "'SNOWFLAKE'",
-        "BASE_LOCATION": "/root",
-        "CATALOG_SYNC": "foo",
-        "STORAGE_SERIALIZATION_POLICY": "bar",
-    }
-    assert (
-        _iceberg_config_statement_helper(config)
-        == "EXTERNAL_VOLUME='vol' CATALOG='SNOWFLAKE' BASE_LOCATION='/root' CATALOG_SYNC='foo' STORAGE_SERIALIZATION_POLICY='bar'"
-    )
-
-    config["STORAGE_SERIALIZATION_POLICY"] = None
-    assert (
-        _iceberg_config_statement_helper(config)
-        == "EXTERNAL_VOLUME='vol' CATALOG='SNOWFLAKE' BASE_LOCATION='/root' CATALOG_SYNC='foo'"
-    )
-
-    config["foo"] = True
-    config["bar"] = True
-    with pytest.raises(
-        ProgrammingError,
-        match=re.escape("Invalid iceberg configurations option(s) provided BAR, FOO"),
-    ):
-        _iceberg_config_statement_helper(config)
-
-
 async def test_write_pandas_with_on_error(
     conn_cnx: Callable[..., Generator[SnowflakeConnection]],
 ):