ENH: Further refinements of interval

Author: bashtage

.flake8

@@ -0,0 +1,3 @@
+[flake8]
+per-file-ignores =
+    tests/test_scalars.py: W503

pandas-stubs/_libs/interval.pyi

@@ -133,9 +133,7 @@ class Interval(IntervalMixin, Generic[_OrderableT]):
     @overload
     def __rmul__(self: Interval[Timedelta], y: float) -> Interval[Timedelta]: ...
     @overload
-    def __truediv__(
-        self: Interval[int], y: _OrderableScalarT
-    ) -> Interval[_OrderableScalarT]: ...
+    def __truediv__(self: Interval[int], y: _OrderableScalarT) -> Interval[float]: ...
     @overload
     def __truediv__(self: Interval[float], y: float) -> Interval[float]: ...
     @overload

pandas-stubs/_typing.pyi

@@ -26,6 +26,7 @@ from pandas.core.indexes.base import Index
 from pandas.core.series import Series
 from typing_extensions import TypeAlias
 
+from pandas._libs.interval import Interval
 from pandas._libs.tslibs import (
     Period,
     Timedelta,
@@ -196,6 +197,7 @@ S1 = TypeVar(
     Timedelta,
     np.datetime64,
     Period,
+    Interval,
 )
 T1 = TypeVar(
     "T1", str, int, np.int64, np.uint64, np.float64, float, np.dtype[np.generic]

pandas-stubs/core/indexes/interval.pyi

@@ -25,6 +25,7 @@ from pandas._typing import (
     FillnaOptions,
     IntervalClosedType,
     Label,
+    np_ndarray_bool,
     npt,
 )
 
@@ -122,6 +123,30 @@ class IntervalIndex(IntervalMixin, ExtensionIndex):
     def get_value(self, series: ABCSeries, key): ...
     @property
     def is_all_dates(self) -> bool: ...
+    @overload  # type: ignore[override]
+    def __gt__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __gt__(self, other: object) -> bool: ...
+    @overload  # type: ignore[override]
+    def __ge__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __ge__(self, other: object) -> bool: ...
+    @overload  # type: ignore[override]
+    def __le__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __le__(self, other: object) -> bool: ...
+    @overload  # type: ignore[override]
+    def __lt__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __lt__(self, other: object) -> bool: ...
+    @overload  # type: ignore[override]
+    def __eq__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __eq__(self, other: object) -> bool: ...
+    @overload  # type: ignore[override]
+    def __ne__(self, other: Interval | IntervalIndex) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __ne__(self, other: object) -> bool: ...
 
 @overload
 def interval_range(

pandas-stubs/core/series.pyi

@@ -24,6 +24,7 @@ from matplotlib.axes import (
 )
 import numpy as np
 from pandas import (
+    Interval,
     Period,
     Timedelta,
     Timestamp,
@@ -43,6 +44,7 @@ from pandas.core.indexes.accessors import (
 )
 from pandas.core.indexes.base import Index
 from pandas.core.indexes.datetimes import DatetimeIndex
+from pandas.core.indexes.interval import IntervalIndex
 from pandas.core.indexes.period import PeriodIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 from pandas.core.indexing import (
@@ -196,6 +198,16 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
         fastpath: bool = ...,
     ) -> Series[Period]: ...
     @overload
+    def __new__(
+        cls,
+        data: IntervalIndex,
+        index: Axes | None = ...,
+        dtype=...,
+        name: Hashable | None = ...,
+        copy: bool = ...,
+        fastpath: bool = ...,
+    ) -> Series[Interval]: ...
+    @overload
     def __new__(
         cls,
         data: object | _ListLike | Series[S1] | dict[int, S1] | dict[_str, S1] | None,