restore DateTimeIndex methods

Author: yangdanny97

pandas-stubs/core/indexes/datetimelike.pyi

@@ -1,10 +1,12 @@
+import numpy as np
 from pandas.core.indexes.extension import ExtensionIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 from typing_extensions import Self
 
 from pandas._libs.tslibs import BaseOffset
 from pandas._typing import (
     S1,
+    AxisIndex,
     TimeUnit,
 )
 
@@ -15,6 +17,18 @@ class DatetimeIndexOpsMixin(ExtensionIndex[S1]):
     def freqstr(self) -> str | None: ...
     @property
     def is_all_dates(self) -> bool: ...
+    def min(
+        self, axis: AxisIndex | None = ..., skipna: bool = ..., *args, **kwargs
+    ) -> S1: ...
+    def argmin(
+        self, axis: AxisIndex | None = ..., skipna: bool = ..., *args, **kwargs
+    ) -> np.int64: ...
+    def max(
+        self, axis: AxisIndex | None = ..., skipna: bool = ..., *args, **kwargs
+    ) -> S1: ...
+    def argmax(
+        self, axis: AxisIndex | None = ..., skipna: bool = ..., *args, **kwargs
+    ) -> np.int64: ...
     def __rsub__(  # type: ignore[override]
         self, other: DatetimeIndexOpsMixin
     ) -> TimedeltaIndex: ...

tests/test_indexes.py

@@ -1344,8 +1344,8 @@ def test_datetime_index_max_min_reductions() -> None:
     dtidx = pd.DatetimeIndex(["2020-01-01", "2020-01-02"])
     check(assert_type(dtidx.argmax(), np.int64), np.int64)
     check(assert_type(dtidx.argmin(), np.int64), np.int64)
-    check(assert_type(dtidx.max(), Any), pd.Timestamp)
-    check(assert_type(dtidx.min(), Any), pd.Timestamp)
+    check(assert_type(dtidx.max(), pd.Timestamp), pd.Timestamp)
+    check(assert_type(dtidx.min(), pd.Timestamp), pd.Timestamp)
 
 
 def test_periodindex_shift() -> None: