feat: support Series.dt.normalize

bigframes/core/compile/scalar_op_compiler.py

@@ -632,6 +632,11 @@ def year_op_impl(x: ibis_types.Value):
     return typing.cast(ibis_types.TimestampValue, x).year().cast(ibis_dtypes.int64)
 
 
+@scalar_op_compiler.register_unary_op(ops.normalize_op)
+def normalize_op_impl(x: ibis_types.Value):
+    return typing.cast(ibis_types.TimestampValue, x).truncate("D")
+
+
 # Parameterized ops
 @scalar_op_compiler.register_unary_op(ops.StructFieldOp, pass_op=True)
 def struct_field_op_impl(x: ibis_types.Value, op: ops.StructFieldOp):

bigframes/operations/__init__.py

@@ -206,6 +206,7 @@ def create_ternary_op(
 second_op = create_unary_op(name="second", type_rule=op_typing.INTEGER)
 time_op = create_unary_op(name="time", type_rule=op_typing.INTEGER)
 year_op = create_unary_op(name="year", type_rule=op_typing.INTEGER)
+normalize_op = create_unary_op(name="normalize")
 ## Trigonometry Ops
 sin_op = create_unary_op(name="sin", type_rule=op_typing.REAL_NUMERIC)
 cos_op = create_unary_op(name="cos", type_rule=op_typing.REAL_NUMERIC)

bigframes/operations/datetimes.py

@@ -94,3 +94,6 @@ def unit(self) -> str:
 
     def strftime(self, date_format: str) -> series.Series:
         return self._apply_unary_op(ops.StrftimeOp(date_format=date_format))
+
+    def normalize(self) -> series.Series:
+        return self._apply_unary_op(ops.normalize_op)

tests/system/small/operations/test_datetimes.py

@@ -266,3 +266,19 @@ def test_dt_strftime_time():
         bf_result, expected_result, check_index_type=False, check_dtype=False
     )
     assert bf_result.dtype == "string[pyarrow]"
+
+
+@pytest.mark.parametrize(
+    ("col_name",),
+    DATETIME_COL_NAMES,
+)
+def test_dt_normalize(scalars_dfs, col_name):
+    scalars_df, scalars_pandas_df = scalars_dfs
+    bf_series: bigframes.series.Series = scalars_df[col_name]
+    bf_result = bf_series.dt.normalize().to_pandas()
+    pd_result = scalars_pandas_df[col_name].dt.normalize()
+
+    assert_series_equal(
+        pd_result.astype(pd.Int64Dtype()),
+        bf_result,
+    )

third_party/bigframes_vendored/pandas/core/arrays/datetimelike.py

@@ -36,3 +36,31 @@ def strftime(self, date_format: str):
             bigframes.series.Series of formatted strings.
         """
         raise NotImplementedError(constants.ABSTRACT_METHOD_ERROR_MESSAGE)
+
+    def normalize(self):
+        """
+        Convert times to midnight.
+
+        The time component of the date-time is converted to midnight i.e.
+        00:00:00. This is useful in cases, when the time does not matter.
+        Length is unaltered. The timezones are unaffected.
+
+        This method is available on Series with datetime values under the
+        .dt accessor.
+
+        **Examples:**
+
+           >>> import pandas as pd
+           >>> impor bigframes.pandas as bpd
+           >>> s = bpd.Series(pd.date_range(
+                       start='2014-08-01 10:00',
+                       freq='h',
+                       periods=3,
+                       tz='Asia/Calcutta')
+                    )
+           >>> s.dt.normalize()
+
+        Returns:
+            bigframes.series.Series of the same dtype as the input.
+        """
+        raise NotImplementedError(constants.ABSTRACT_METHOD_ERROR_MESSAGE)