More tests

Author: jprakash-db

tests/test_local/test_ddl.py

@@ -1,13 +1,13 @@
 import pytest
-from sqlalchemy import Column, MetaData, String, Table, create_engine
+from sqlalchemy import Column, MetaData, String, Table, Numeric, create_engine
 from sqlalchemy.schema import (
     CreateTable,
     DropColumnComment,
     DropTableComment,
     SetColumnComment,
     SetTableComment,
 )
-
+from databricks.sqlalchemy import DatabricksArray,DatabricksMap
 
 class DDLTestBase:
     engine = create_engine(
@@ -94,3 +94,22 @@ def test_alter_table_drop_comment(self, table_with_comment):
         stmt = DropTableComment(table_with_comment)
         output = self.compile(stmt)
         assert output == "COMMENT ON TABLE martin IS NULL"
+
+class TestTableComplexTypeDDL(DDLTestBase):
+    @pytest.fixture
+    def metadata(self) -> MetaData:
+        metadata = MetaData()
+        col1 = Column("array_array_string",DatabricksArray(DatabricksArray(String)))
+        col2 = Column("map_string_string",DatabricksMap(String,String))
+        col3 = Column("array_array_decimal",DatabricksArray(DatabricksArray(Numeric(10,2))))
+        table = Table("complex_type", metadata, col1,col2,col3)
+        return metadata
+    
+    def test_create_table_with_complex_type(self, metadata):
+        stmt = CreateTable(metadata.tables["complex_type"])
+        output = self.compile(stmt)
+
+        print(output)
+        assert "array_array_string ARRAY<ARRAY<STRING>>" in output
+        assert "map_string_string MAP<STRING,STRING>" in output
+        assert "array_array_decimal ARRAY<ARRAY<DECIMAL(10,2)>>" in output