TST Add tests for dtypes

Author: Kevin Sheppard

pandas-stubs/_libs/interval.pyi

@@ -63,8 +63,10 @@ class Interval(IntervalMixin, Generic[_OrderableT]):
     def right(self: Interval[_OrderableT]) -> _OrderableT: ...
     @property
     def closed(self) -> IntervalClosedType: ...
-    mid: _MidDescriptor
-    length: _LengthDescriptor
+    @property
+    def mid(self) -> _MidDescriptor: ...
+    @property
+    def length(self) -> _LengthDescriptor: ...
     def __init__(
         self,
         left: _OrderableT,
@@ -142,10 +144,6 @@ class Interval(IntervalMixin, Generic[_OrderableT]):
     def __floordiv__(self: Interval[float], y: float) -> Interval[float]: ...
     def overlaps(self: Interval[_OrderableT], other: Interval[_OrderableT]) -> bool: ...
 
-def intervals_to_interval_bounds(
-    intervals: np.ndarray, validate_closed: bool = ...
-) -> tuple[np.ndarray, np.ndarray, str]: ...
-
 class IntervalTree(IntervalMixin):
     def __init__(
         self,

tests/test_dtypes.py

@@ -0,0 +1,109 @@
+from datetime import (
+    timedelta,
+    timezone,
+)
+
+import numpy as np
+import pandas as pd
+from typing_extensions import assert_type
+
+from tests import check
+
+from pandas.tseries.offsets import (
+    BusinessDay,
+    CustomBusinessDay,
+    Day,
+)
+
+
+def test_datetimetz_dtype() -> None:
+    check(
+        assert_type(pd.DatetimeTZDtype(unit="ns", tz="UTC"), pd.DatetimeTZDtype),
+        pd.DatetimeTZDtype,
+    )
+    check(
+        assert_type(
+            pd.DatetimeTZDtype(unit="ns", tz=timezone(timedelta(hours=1))),
+            pd.DatetimeTZDtype,
+        ),
+        pd.DatetimeTZDtype,
+    )
+
+
+def test_period_dtype() -> None:
+    check(assert_type(pd.PeriodDtype(freq="D"), pd.PeriodDtype), pd.PeriodDtype)
+    check(assert_type(pd.PeriodDtype(freq=Day()), pd.PeriodDtype), pd.PeriodDtype)
+    check(
+        assert_type(pd.PeriodDtype(freq=BusinessDay()), pd.PeriodDtype), pd.PeriodDtype
+    )
+    check(
+        assert_type(pd.PeriodDtype(freq=CustomBusinessDay()), pd.PeriodDtype),
+        pd.PeriodDtype,
+    )
+
+
+def test_interval_dtype() -> None:
+    check(
+        assert_type(
+            pd.Interval(pd.Timestamp("2017-01-01"), pd.Timestamp("2017-01-02")),
+            pd.Interval,
+        ),
+        pd.Interval,
+    )
+    check(assert_type(pd.Interval(1, 2, closed="left"), pd.Interval), pd.Interval)
+    check(assert_type(pd.Interval(1.0, 2.5, closed="right"), pd.Interval), pd.Interval)
+    check(assert_type(pd.Interval(1.0, 2.5, closed="both"), pd.Interval), pd.Interval)
+    check(
+        assert_type(pd.Interval(1.0, 2.5, closed="neither"), pd.Interval), pd.Interval
+    )
+    check(
+        assert_type(
+            pd.Interval(pd.Timedelta("1 days"), pd.Timedelta("2 days")), pd.Interval
+        ),
+        pd.Interval,
+    )
+
+
+def test_int64_dtype() -> None:
+    check(assert_type(pd.Int64Dtype(), pd.Int64Dtype), pd.Int64Dtype)
+
+
+def test_categorical_dtype() -> None:
+    check(
+        assert_type(
+            pd.CategoricalDtype(categories=["a", "b", "c"], ordered=True),
+            pd.CategoricalDtype,
+        ),
+        pd.CategoricalDtype,
+    )
+    check(
+        assert_type(pd.CategoricalDtype(categories=[1, 2, 3]), pd.CategoricalDtype),
+        pd.CategoricalDtype,
+    )
+
+
+def test_sparse_dtype() -> None:
+    check(assert_type(pd.SparseDtype(str), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(complex), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(bool), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(int), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(np.int64), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(str), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(float), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(np.datetime64), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(np.timedelta64), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype("datetime64"), pd.SparseDtype), pd.SparseDtype)
+    check(assert_type(pd.SparseDtype(), pd.SparseDtype), pd.SparseDtype)
+
+
+def test_string_dtype() -> None:
+    check(assert_type(pd.StringDtype("pyarrow"), pd.StringDtype), pd.StringDtype)
+    check(assert_type(pd.StringDtype("python"), pd.StringDtype), pd.StringDtype)
+
+
+def test_boolean_dtype() -> None:
+    check(assert_type(pd.BooleanDtype(), pd.BooleanDtype), pd.BooleanDtype)
+
+
+def test_arrow_dtype() -> None:
+    check(assert_type(pd.ArrowDtype("int64"), pd.ArrowDtype), pd.ArrowDtype)