Merge pull request #4 from HagonChan/AutoCodeRover-#1-20241004_10_01_35

Fix #1 (AutoCodeRover)

Author: HagonChan

pandas/core/arrays/categorical.py

@@ -549,7 +549,7 @@ def astype(self, dtype: ExtensionDtype, copy: bool = ...) -> ExtensionArray: ...
     @overload
     def astype(self, dtype: AstypeArg, copy: bool = ...) -> ArrayLike: ...
 
-    def astype(self, dtype: AstypeArg, copy: bool = True) -> ArrayLike:
+    def astype(self, dtype: AstypeArg, copy: bool = True, errors: str = "raise") -> ArrayLike:
         """
         Coerce this type to another dtype
 
@@ -560,6 +560,11 @@ def astype(self, dtype: AstypeArg, copy: bool = True) -> ArrayLike:
             By default, astype always returns a newly allocated object.
             If copy is set to False and dtype is categorical, the original
             object is returned.
+        errors : {'raise', 'ignore'}, default 'raise'
+            Control raising of exceptions on invalid data for provided dtype.
+
+            - 'raise' : allow exceptions to be raised
+            - 'ignore' : suppress exceptions. On error return original object
         """
         dtype = pandas_dtype(dtype)
         result: Categorical | np.ndarray
@@ -603,6 +608,9 @@ def astype(self, dtype: AstypeArg, copy: bool = True) -> ArrayLike:
                 msg = f"Cannot cast {self.categories.dtype} dtype to {dtype}"
                 raise ValueError(msg) from err
 
+            if errors == "raise" and not np.all(np.isin(self.codes, new_cats)):
+                raise ValueError("Cannot convert to CategoricalDtype with undefined values")
+
             result = take_nd(
                 new_cats, ensure_platform_int(self._codes), fill_value=fill_value
             )