MAINT: Merge main

Author: bashtage

pandas-stubs/_testing/__init__.pyi

@@ -1,4 +1,9 @@
-from typing import Literal
+from contextlib import contextmanager
+from typing import (
+    Any,
+    Generator,
+    Literal,
+)
 
 from pandas import (
     DataFrame,
@@ -107,3 +112,7 @@ def assert_produces_warning(
     check_stacklevel: bool = ...,
     raise_on_extra_warnings: bool = ...,
 ) -> None: ...
+@contextmanager
+def ensure_clean(
+    filename: str | None = ..., return_filelike: bool = ..., **kwargs: Any
+) -> Generator[str, None, None]: ...

pandas-stubs/_typing.pyi

@@ -220,4 +220,12 @@ StataDateFormat = Literal[
     "%ty",
 ]
 
+FillnaOptions = Literal["backfill", "bfill", "ffill", "pad"]
+ReplaceMethod = Literal["pad", "ffill", "bfill"]
+SortKind = Literal["quicksort", "mergesort", "heapsort", "stable"]
+NaPosition = Literal["first", "last"]
+MergeHow = Literal["left", "right", "outer", "inner"]
+JsonOrient = Literal["split", "records", "index", "columns", "values", "table"]
+TimestampConvention = Literal["start", "end", "s", "e"]
+
 __all__ = ["npt", "type_t"]

pandas-stubs/_version.pyi

@@ -4,4 +4,4 @@ version_json: str = ...
 
 def get_versions(): ...
 
-_stub_version = Literal["1.4.3.220815"]
+_stub_version = Literal["1.4.3.220822"]

pandas-stubs/core/base.pyi

@@ -10,6 +10,7 @@ from pandas.core.arrays import ExtensionArray
 from pandas.core.arrays.categorical import Categorical
 
 from pandas._typing import (
+    NaPosition,
     NDFrameT,
     SeriesAxisType,
 )
@@ -75,7 +76,9 @@ class IndexOpsMixin:
     def factorize(
         self, sort: bool = ..., na_sentinel: int = ...
     ) -> tuple[np.ndarray, np.ndarray | Index | Categorical]: ...
-    def searchsorted(self, value, side: str = ..., sorter=...) -> int | list[int]: ...
+    def searchsorted(
+        self, value, side: Literal["left", "right"] = ..., sorter=...
+    ) -> int | list[int]: ...
     def drop_duplicates(
-        self, keep: Literal["first", "last", False] = ...
+        self, keep: NaPosition | Literal[False] = ...
     ) -> IndexOpsMixin: ...