misc

Author: cmp0xff

pandas-stubs/_testing/__init__.pyi

@@ -11,7 +11,6 @@ from typing import (
 )
 import warnings
 
-from matplotlib.artist import Artist
 from pandas import (
     Categorical,
     DataFrame,
@@ -60,9 +59,6 @@ def assert_class_equal(
 def assert_attr_equal(
     attr: str, left: object, right: object, obj: str = "Attributes"
 ) -> None: ...
-def assert_is_valid_plot_return_object(
-    objs: Series | np_ndarray | Artist | tuple | dict,
-) -> None: ...
 def assert_is_sorted(seq: AnyArrayLike) -> None: ...
 def assert_categorical_equal(
     left: Categorical,

pandas-stubs/core/algorithms.pyi

@@ -42,7 +42,7 @@ def unique(values: PeriodIndex) -> PeriodIndex: ...
 # switch to DatetimeIndex after Pandas 3.0
 def unique(values: DatetimeIndex) -> np_1darray_dt | DatetimeIndex: ...
 @overload
-# switch to DatetimeIndex after Pandas 3.0
+# switch to TimedeltaIndex after Pandas 3.0
 def unique(values: TimedeltaIndex) -> np_1darray_td: ...
 @overload
 # switch to Index[int] after Pandas 3.0

pandas-stubs/core/reshape/tile.pyi

@@ -225,7 +225,7 @@ def qcut(
     retbins: Literal[True],
     precision: int = 3,
     duplicates: Literal["raise", "drop"] = "raise",
-    # double when there are nan's
+    # float when there are nan's
 ) -> tuple[np_1darray_intp | np_1darray_float, np_1darray_float]: ...
 @overload
 def qcut(

tests/test_pandas.py

@@ -967,7 +967,7 @@ def test_index_unqiue() -> None:
     )
     check(
         assert_type(pd.unique(tdi), np_1darray_td),
-        np_1darray if PD_LTE_23 else pd.Index,
+        np_1darray if PD_LTE_23 else pd.TimedeltaIndex,
     )
     check(
         assert_type(pd.unique(mi), np_ndarray),