more tests

Author: MarcoGorelli

pandas-stubs/core/indexes/base.pyi

@@ -524,7 +524,7 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
         self, start: SliceType = None, end: SliceType = None, step: int | None = None
     ): ...
     def delete(self, loc: int | AnyArrayLikeInt | Sequence[int]) -> Self: ...
-    def insert(self, loc: int, item: S1) -> Self: ...
+    def insert(self, loc: int, item: object) -> Index: ...
     def drop(self, labels, errors: IgnoreRaise = "raise") -> Self: ...
     @property
     def shape(self) -> tuple[int, ...]: ...

tests/indexes/test_categoricalindex.py

@@ -25,3 +25,16 @@ def test_categoricalindex_reindex() -> None:
     reindexed = ci.reindex(["b", "c"])
     check(assert_type(reindexed[0], pd.Index), pd.Index)
     check(assert_type(reindexed[1], np_1darray[np.intp] | None), np_1darray)
+
+
+def test_categoricalindex_delete() -> None:
+    ci = pd.CategoricalIndex(["a", "b"])
+    check(assert_type(ci.delete(0), "pd.CategoricalIndex[str]"), pd.CategoricalIndex)
+    check(
+        assert_type(ci.delete([0, 1]), "pd.CategoricalIndex[str]"), pd.CategoricalIndex
+    )
+
+
+def test_categoricalindex_insert() -> None:
+    ci = pd.CategoricalIndex(["a", "b"])
+    check(assert_type(ci.insert(0, "c"), pd.Index), pd.Index)