SNOW-1871582 Fix support for SA array (#568)

* Fix drop support for SA array

* Update DESCRIPTION.md

Author: sfc-gh-jvasquezrojas

DESCRIPTION.md

@@ -9,6 +9,7 @@ Source code is also available at:
 
 # Release Notes
 - (Unreleased)
+  - Fix support for SqlAlchemy ARRAY.
   - Fix return value of snowflake get_table_names.
   - Fix incorrect quoting of identifiers with `_` as initial character.
   - Added `force_div_is_floordiv` flag to override `div_is_floordiv` new default value `False` in `SnowflakeDialect`.

src/snowflake/sqlalchemy/base.py

@@ -1133,6 +1133,9 @@ def visit_MAP(self, type_, **kw):
         )
 
     def visit_ARRAY(self, type_, **kw):
+        return "ARRAY"
+
+    def visit_SNOWFLAKE_ARRAY(self, type_, **kw):
         if type_.is_semi_structured:
             return "ARRAY"
         not_null = f" {NOT_NULL}" if type_.not_null else ""

src/snowflake/sqlalchemy/custom_types.py

@@ -83,7 +83,7 @@ def __repr__(self):
 
 
 class ARRAY(StructuredType):
-    __visit_name__ = "ARRAY"
+    __visit_name__ = "SNOWFLAKE_ARRAY"
 
     def __init__(
         self,

tests/__snapshots__/test_structured_datatypes.ambr

@@ -5,6 +5,9 @@
 # name: test_compile_table_with_double_map
   'CREATE TABLE clustered_user (\t"Id" INTEGER NOT NULL AUTOINCREMENT, \tname MAP(DECIMAL, MAP(DECIMAL, VARCHAR)), \tPRIMARY KEY ("Id"))'
 # ---
+# name: test_compile_table_with_sqlalchemy_array
+  'CREATE TABLE clustered_user (\t"Id" INTEGER NOT NULL AUTOINCREMENT, \tname ARRAY, \tPRIMARY KEY ("Id"))'
+# ---
 # name: test_compile_table_with_structured_data_type[structured_type0]
   'CREATE TABLE clustered_user (\t"Id" INTEGER NOT NULL AUTOINCREMENT, \tname MAP(DECIMAL(10, 0), MAP(DECIMAL(10, 0), VARCHAR(16777216))), \tPRIMARY KEY ("Id"))'
 # ---

tests/test_structured_datatypes.py

@@ -2,6 +2,7 @@
 # Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
 #
 import pytest
+import sqlalchemy as sa
 from sqlalchemy import (
     Column,
     Integer,
@@ -47,6 +48,20 @@ def test_compile_table_with_structured_data_type(
     assert sql_compiler(create_table) == snapshot
 
 
+def test_compile_table_with_sqlalchemy_array(sql_compiler, snapshot):
+    metadata = MetaData()
+    user_table = Table(
+        "clustered_user",
+        metadata,
+        Column("Id", Integer, primary_key=True),
+        Column("name", sa.ARRAY(sa.String)),
+    )
+
+    create_table = CreateTable(user_table)
+
+    assert sql_compiler(create_table) == snapshot
+
+
 @pytest.mark.requires_external_volume
 def test_insert_map(engine_testaccount, external_volume, base_location, snapshot):
     metadata = MetaData()