rebase to master

Author: rgkimball

pandas/_typing.py

@@ -123,8 +123,8 @@
 Frequency = Union[str, "DateOffset"]
 Axes = Collection[Any]
 RandomState = Union[int, ArrayLike, np.random.Generator, np.random.RandomState]
-MergeTypes = Literal['inner', 'outer', 'left', 'right', 'cross']
-ConcatTypes = Literal['inner', 'outer']
+MergeTypes = Literal["inner", "outer", "left", "right", "cross"]
+ConcatTypes = Literal["inner", "outer"]
 
 # dtypes
 NpDtype = Union[str, np.dtype]

pandas/core/apply.py

@@ -520,7 +520,7 @@ def apply_multiple(self) -> FrameOrSeriesUnion:
 
     def normalize_dictlike_arg(
         self,
-        how: Literal['apply', 'agg', 'transform'],
+        how: Literal["apply", "agg", "transform"],
         obj: FrameOrSeriesUnion,
         func: AggFuncTypeDict,
     ) -> AggFuncTypeDict:

pandas/core/arrays/_ranges.py

@@ -79,7 +79,7 @@ def generate_regular_range(
 
 
 def _generate_range_overflow_safe(
-    endpoint: int, periods: int, stride: int, side: Literal['start', 'end'] = "start"
+    endpoint: int, periods: int, stride: int, side: Literal["start", "end"] = "start"
 ) -> int:
     """
     Calculate the second endpoint for passing to np.arange, checking
@@ -106,8 +106,7 @@ def _generate_range_overflow_safe(
     OutOfBoundsDatetime
     """
     # GH#14187 raise instead of incorrectly wrapping around
-    if side not in ["start", "end"]:
-        raise ValueError('Value for side argument must be one of: start, end')
+    assert side in ["start", "end"]
 
     i64max = np.uint64(i8max)
     msg = f"Cannot generate range with {side}={endpoint} and periods={periods}"
@@ -147,14 +146,14 @@ def _generate_range_overflow_safe(
 
 
 def _generate_range_overflow_safe_signed(
-    endpoint: int, periods: int, stride: int, side: Literal['start', 'end']
+    endpoint: int, periods: int, stride: int, side: Literal["start", "end"]
 ) -> int:
     """
     A special case for _generate_range_overflow_safe where `periods * stride`
     can be calculated without overflowing int64 bounds.
     """
-    if side not in ['start', 'end']:
-        raise ValueError('Value for side argument must be one of: start, end')
+    assert side in ['start', 'end']
+
     if side == 'end':
         stride *= -1
 

pandas/core/arrays/datetimes.py

@@ -19,6 +19,9 @@
     lib,
     tslib,
 )
+from pandas._typing import (
+    Dtype,
+)
 from pandas._libs.arrays import NDArrayBacked
 from pandas._libs.tslibs import (
     BaseOffset,
@@ -1967,7 +1970,7 @@ def sequence_to_datetimes(
 
 def sequence_to_dt64ns(
     data,
-    dtype: str = None,
+    dtype: Dtype = None,
     copy: bool = False,
     tz: tzinfo | str = None,
     dayfirst: bool = False,

pandas/core/groupby/ops.py

@@ -967,9 +967,7 @@ def _cython_operation(
         """
         Returns the values of a cython operation.
         """
-        if kind not in ["transform", "aggregate"]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'transform, aggregate')
+        assert kind in ["transform", "aggregate"]
 
         cy_op = WrappedCythonOp(kind=kind, how=how)
 

pandas/core/indexes/base.py

@@ -3651,8 +3651,7 @@ def _convert_slice_indexer(self, key: slice, kind: str_t):
         key : label of the slice bound
         kind : {'loc', 'getitem'}
         """
-        if kind not in ["loc", "getitem"]:
-            raise ValueError('Value for kind argument must be one of: loc, getitem')
+        assert kind in ["loc", "getitem"], kind
 
         # potentially cast the bounds to integers
         start, stop, step = key.start, key.stop, key.step
@@ -5679,8 +5678,7 @@ def _validate_indexer(self, form: str_t, key, kind: str_t):
         If we are positional indexer, validate that we have appropriate
         typed bounds must be an integer.
         """
-        if kind not in ["getitem", "iloc"]:
-            raise ValueError('Value for kind argument must be one of: getitem, iloc')
+        assert kind in ["getitem", "iloc"]
 
         if key is not None and not is_integer(key):
             raise self._invalid_indexer(form, key)
@@ -5710,9 +5708,7 @@ def _maybe_cast_slice_bound(
         -----
         Value of `side` parameter should be validated in caller.
         """
-        if kind not in ["loc", "getitem", None, no_default]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'loc, getitem or None')
+        assert kind in ["loc", "getitem", None, no_default]
         self._deprecated_arg(kind, "kind", "_maybe_cast_slice_bound")
 
         # We are a plain index here (sub-class override this method if they

pandas/core/indexes/datetimes.py

@@ -728,9 +728,7 @@ def _maybe_cast_slice_bound(self, label, side: str, kind=lib.no_default):
         -----
         Value of `side` parameter should be validated in caller.
         """
-        if kind not in ["loc", "getitem", None, lib.no_default]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'loc, getitem or None')
+        assert kind in ["loc", "getitem", None, lib.no_default]
         self._deprecated_arg(kind, "kind", "_maybe_cast_slice_bound")
 
         if isinstance(label, str):

pandas/core/indexes/numeric.py

@@ -236,8 +236,7 @@ def _should_fallback_to_positional(self) -> bool:
     @doc(Index._convert_slice_indexer)
     def _convert_slice_indexer(self, key: slice, kind: str):
         if is_float_dtype(self.dtype):
-            if kind not in ["loc", "getitem"]:
-                raise ValueError('Value for kind argument must be one of: loc, getitem')
+            assert kind in ["loc", "getitem"]
 
             # We always treat __getitem__ slicing as label-based
             # translate to locations
@@ -247,9 +246,7 @@ def _convert_slice_indexer(self, key: slice, kind: str):
 
     @doc(Index._maybe_cast_slice_bound)
     def _maybe_cast_slice_bound(self, label, side: str, kind=lib.no_default):
-        if kind not in ["loc", "getitem", None, lib.no_default]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'loc, getitem or None')
+        assert kind in ["loc", "getitem", None, lib.no_default]
         self._deprecated_arg(kind, "kind", "_maybe_cast_slice_bound")
 
         # we will try to coerce to integers

pandas/core/indexes/period.py

@@ -480,9 +480,7 @@ def _maybe_cast_slice_bound(self, label, side: str, kind=lib.no_default):
         Value of `side` parameter should be validated in caller.
 
         """
-        if kind not in ["loc", "getitem", None, lib.no_default]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'loc, getitem or None')
+        assert kind in ["loc", "getitem", None, lib.no_default]
         self._deprecated_arg(kind, "kind", "_maybe_cast_slice_bound")
 
         if isinstance(label, datetime):

pandas/core/indexes/timedeltas.py

@@ -192,9 +192,7 @@ def _maybe_cast_slice_bound(self, label, side: str, kind=lib.no_default):
         -------
         label : object
         """
-        if kind not in ["loc", "getitem", None, lib.no_default]:
-            raise ValueError('Value for kind argument must be one of: '
-                             'loc, getitem or None')
+        assert kind in ["loc", "getitem", None, lib.no_default]
         self._deprecated_arg(kind, "kind", "_maybe_cast_slice_bound")
 
         if isinstance(label, str):

pandas/core/missing.py

@@ -165,7 +165,7 @@ def clean_interp_method(method: str, index: Index, **kwargs) -> str:
     return method
 
 
-def find_valid_index(values, *, how: Literal['first', 'last']) -> int | None:
+def find_valid_index(values, *, how: Literal["first", "last"]) -> int | None:
     """
     Retrieves the index of the first valid value.