fix

Author: jialuoo

tests/unit/core/compile/sqlglot/expressions/snapshots/test_datetime_ops/test_datetime_to_integer_label/out.sql

@@ -12,24 +12,6 @@ WITH `bfcte_0` AS (
         86400000000
       )
     ) AS INT64) AS `bfcol_2`,
-    CASE
-      WHEN (
-        EXTRACT(YEAR FROM `datetime_col`) * 12 + EXTRACT(MONTH FROM `datetime_col`) - 1
-      ) = (
-        EXTRACT(YEAR FROM `timestamp_col`) * 12 + EXTRACT(MONTH FROM `timestamp_col`) - 1
-      )
-      THEN 0
-      ELSE CAST(FLOOR(
-        IEEE_DIVIDE(
-          (
-            EXTRACT(YEAR FROM `datetime_col`) * 12 + EXTRACT(MONTH FROM `datetime_col`) - 1
-          ) - (
-            EXTRACT(YEAR FROM `timestamp_col`) * 12 + EXTRACT(MONTH FROM `timestamp_col`) - 1
-          ) - 1,
-          1
-        )
-      ) AS INT64) + 1
-    END AS `bfcol_3`,
     CASE
       WHEN UNIX_MICROS(
         CAST(TIMESTAMP_TRUNC(`datetime_col`, WEEK(MONDAY)) + INTERVAL 6 DAY AS TIMESTAMP)
@@ -47,38 +29,10 @@ WITH `bfcte_0` AS (
           604800000000
         )
       ) AS INT64) + 1
-    END AS `bfcol_4`,
-    CASE
-      WHEN (
-        EXTRACT(YEAR FROM `datetime_col`) * 4 + EXTRACT(QUARTER FROM `datetime_col`) - 1
-      ) = (
-        EXTRACT(YEAR FROM `timestamp_col`) * 4 + EXTRACT(QUARTER FROM `timestamp_col`) - 1
-      )
-      THEN 0
-      ELSE CAST(FLOOR(
-        IEEE_DIVIDE(
-          (
-            EXTRACT(YEAR FROM `datetime_col`) * 4 + EXTRACT(QUARTER FROM `datetime_col`) - 1
-          ) - (
-            EXTRACT(YEAR FROM `timestamp_col`) * 4 + EXTRACT(QUARTER FROM `timestamp_col`) - 1
-          ) - 1,
-          1
-        )
-      ) AS INT64) + 1
-    END AS `bfcol_5`,
-    CASE
-      WHEN EXTRACT(YEAR FROM `datetime_col`) = EXTRACT(YEAR FROM `timestamp_col`)
-      THEN 0
-      ELSE CAST(FLOOR(
-        IEEE_DIVIDE(EXTRACT(YEAR FROM `datetime_col`) - EXTRACT(YEAR FROM `timestamp_col`) - 1, 1)
-      ) AS INT64) + 1
-    END AS `bfcol_6`
+    END AS `bfcol_3`
   FROM `bfcte_0`
 )
 SELECT
   `bfcol_2` AS `fixed_freq`,
-  `bfcol_3` AS `non_fixed_freq_monthly`,
-  `bfcol_4` AS `non_fixed_freq_weekly`,
-  `bfcol_5` AS `non_fixed_freq_quarterly`,
-  `bfcol_6` AS `non_fixed_freq_yearly`
+  `bfcol_3` AS `non_fixed_freq_weekly`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/test_datetime_ops.py

@@ -64,20 +64,9 @@ def test_datetime_to_integer_label(scalar_types_df: bpd.DataFrame, snapshot):
         "fixed_freq": ops.DatetimeToIntegerLabelOp(
             freq=pd.tseries.offsets.Day(), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
-        "non_fixed_freq_monthly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.MonthEnd(), origin="start", closed="left"  # type: ignore
-        ).as_expr("datetime_col", "timestamp_col"),
         "non_fixed_freq_weekly": ops.DatetimeToIntegerLabelOp(
             freq=pd.tseries.offsets.Week(weekday=6), origin="start", closed="left"  # type: ignore
         ).as_expr("datetime_col", "timestamp_col"),
-        "non_fixed_freq_quarterly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.QuarterEnd(startingMonth=12),  # type: ignore
-            origin="start",
-            closed="left",
-        ).as_expr("datetime_col", "timestamp_col"),
-        "non_fixed_freq_yearly": ops.DatetimeToIntegerLabelOp(
-            freq=pd.tseries.offsets.YearEnd(month=12), origin="start", closed="left"  # type: ignore
-        ).as_expr("datetime_col", "timestamp_col"),
     }
 
     sql = utils._apply_ops_to_sql(bf_df, list(ops_map.values()), list(ops_map.keys()))