minor test updates

Author: fivetran-kwoodbeck

tests/dialects/test_snowflake.py

@@ -1946,9 +1946,9 @@ def test_snowflake(self):
             },
         )
         self.validate_all(
-            "SELECT DATE_PART(NANOSECOND, CAST('2026-01-06 11:45:00.123456789' AS TIMESTAMP))",
+            "SELECT DATE_PART(NANOSECOND, CAST('2026-01-06 11:45:00.123456789' AS TIMESTAMPNTZ))",
             write={
-                "snowflake": "SELECT DATE_PART(NANOSECOND, CAST('2026-01-06 11:45:00.123456789' AS TIMESTAMP))",
+                "snowflake": "SELECT DATE_PART(NANOSECOND, CAST('2026-01-06 11:45:00.123456789' AS TIMESTAMPNTZ))",
                 "duckdb": "SELECT CAST(STRFTIME(CAST(CAST('2026-01-06 11:45:00.123456789' AS TIMESTAMP) AS TIMESTAMP_NS), '%n') AS BIGINT)",
             },
         )
@@ -2037,6 +2037,13 @@ def test_snowflake(self):
                 "duckdb": "SELECT CAST(STRFTIME(CAST('2026-01-06 11:45:00' AS TIMESTAMP), '%G') AS INT)",
             },
         )
+        self.validate_all(
+            "SELECT EXTRACT(HOUR FROM CAST('2026-01-06 11:45:00' AS TIMESTAMP_NTZ))",
+            write={
+                "snowflake": "SELECT DATE_PART(HOUR, CAST('2026-01-06 11:45:00' AS TIMESTAMPNTZ))",
+                "duckdb": "SELECT EXTRACT(HOUR FROM CAST('2026-01-06 11:45:00' AS TIMESTAMP))",
+            },
+        )
         self.validate_all(
             "SELECT EXTRACT(MINUTE FROM CAST('2026-01-06 11:45:00' AS TIMESTAMP_NTZ))",
             write={
@@ -2171,13 +2178,6 @@ def test_snowflake(self):
                 "duckdb": "SELECT CAST(STRFTIME(CAST('2026-01-06' AS DATE), '%G') AS INT)",
             },
         )
-        self.validate_all(
-            "SELECT EXTRACT(EPOCH_SECOND FROM CAST('2026-01-06' AS DATE))",
-            write={
-                "snowflake": "SELECT DATE_PART(EPOCH_SECOND, CAST('2026-01-06' AS DATE))",
-                "duckdb": "SELECT CAST(EPOCH(CAST('2026-01-06' AS DATE)) AS BIGINT)",
-            },
-        )
         self.validate_all(
             "SELECT EXTRACT(HOUR FROM CAST('11:45:00.123456789' AS TIME))",
             write={
@@ -2387,19 +2387,6 @@ def test_snowflake(self):
             },
         )
 
-        # EXTRACT - converts to DATE_PART in Snowflake
-        self.validate_identity(
-            "SELECT EXTRACT(YEAR FROM CAST('2024-05-09' AS DATE))",
-            "SELECT DATE_PART(YEAR, CAST('2024-05-09' AS DATE))",
-        )
-        self.validate_identity(
-            "SELECT EXTRACT(MONTH FROM CAST('2024-05-09 08:50:57' AS TIMESTAMP))",
-            "SELECT DATE_PART(MONTH, CAST('2024-05-09 08:50:57' AS TIMESTAMP))",
-        )
-        self.validate_identity(
-            "SELECT EXTRACT(MINUTE, CAST('08:50:57' AS TIME))",
-            "SELECT DATE_PART(MINUTE, CAST('08:50:57' AS TIME))",
-        )
         self.validate_identity("SELECT HOUR(CAST('08:50:57' AS TIME))")
         self.validate_identity("SELECT MINUTE(CAST('08:50:57' AS TIME))")
         self.validate_identity("SELECT SECOND(CAST('08:50:57' AS TIME))")

tests/fixtures/optimizer/annotate_functions.sql

@@ -2871,10 +2871,6 @@ INT;
 EXTRACT(YEAROFWEEKISO FROM CAST('2026-01-06' AS DATE));
 INT;
 
-# dialect: snowflake
-EXTRACT(EPOCH_SECOND FROM CAST('2026-01-06' AS DATE));
-BIGINT;
-
 # dialect: snowflake
 EXTRACT(HOUR FROM CAST('11:45:00.123456789' AS TIME));
 INT;