Minor fix

databricks · Sep 4, 2024 · d796304 · d796304
1 parent cf2f4a0
commit d796304
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/databricks/sqlalchemy/test_local/test_types.py b/src/databricks/sqlalchemy/test_local/test_types.py
@@ -52,7 +52,6 @@ class DatabricksDataType(enum.Enum):
     sqlalchemy.types.Unicode: DatabricksDataType.STRING,
     sqlalchemy.types.UnicodeText: DatabricksDataType.STRING,
     sqlalchemy.types.Uuid: DatabricksDataType.STRING,
-    sqlalchemy.types.JSON: DatabricksDataType.STRING
 }
 
 
@@ -129,6 +128,7 @@ def test_numeric_renders_as_decimal_with_precision_and_scale(self):
     sqlalchemy.types.INT: DatabricksDataType.INT,
     sqlalchemy.types.SMALLINT: DatabricksDataType.SMALLINT,
     sqlalchemy.types.TIMESTAMP: DatabricksDataType.TIMESTAMP,
+    sqlalchemy.types.JSON: DatabricksDataType.STRING,
     TINYINT: DatabricksDataType.TINYINT,
     TIMESTAMP: DatabricksDataType.TIMESTAMP,
     TIMESTAMP_NTZ: DatabricksDataType.TIMESTAMP_NTZ,