GH828 Add dict

Author: loicdiridollou

pandas-stubs/core/indexes/datetimes.pyi

@@ -10,6 +10,7 @@ from datetime import (
 from typing import overload
 
 from _typing import (
+    Axes,
     Frequency,
     TimeZones,
 )
@@ -44,7 +45,7 @@ from pandas.tseries.offsets import BaseOffset
 class DatetimeIndex(DatetimeTimedeltaMixin[Timestamp], DatetimeIndexProperties):
     def __init__(
         self,
-        data: AnyArrayLike | list | tuple,
+        data: Axes | AnyArrayLike,
         freq: Frequency = ...,
         tz: TimeZones = ...,
         ambiguous: str = ...,

pandas-stubs/core/indexes/timedeltas.pyi

@@ -28,6 +28,7 @@ from pandas._libs import (
 from pandas._libs.tslibs import BaseOffset
 from pandas._typing import (
     AnyArrayLike,
+    Axes,
     TimedeltaConvertibleTypes,
     num,
 )
@@ -37,8 +38,8 @@ class TimedeltaIndex(DatetimeTimedeltaMixin[Timedelta], TimedeltaIndexProperties
         cls,
         data: (
             AnyArrayLike
-            | list[str]
             | Sequence[dt.timedelta | Timedelta | np.timedelta64 | float]
+            | Axes
         ) = ...,
         freq: str | BaseOffset = ...,
         closed: object = ...,

tests/test_indexes.py

@@ -1324,9 +1324,22 @@ def test_index_delete() -> None:
 
 
 def test_index_dict() -> None:
-    if TYPE_CHECKING_INVALID_USAGE:
-        pd.DatetimeIndex(
-            {  # type: ignore[arg-type]
-                "Jan. 1, 2008": "New Year’s Day"  # pyright: ignore[reportArgumentType]
-            }
-        )
+    """Test passing an ordered iterables to Index and subclasses constructor GH828."""
+    check(
+        assert_type(pd.Index({"Jan. 1, 2008": "New Year’s Day"}), "pd.Index[str]"),
+        pd.Index,
+        str,
+    )
+    check(
+        assert_type(
+            pd.DatetimeIndex({"Jan. 1, 2008": "New Year’s Day"}), pd.DatetimeIndex
+        ),
+        pd.DatetimeIndex,
+    )
+    check(
+        assert_type(
+            pd.TimedeltaIndex({pd.Timedelta(days=1): "New Year’s Day"}),
+            pd.TimedeltaIndex,
+        ),
+        pd.TimedeltaIndex,
+    )