pre-commit changes #1

Author: ZanirP

pandas/core/generic.py

@@ -121,10 +121,10 @@
     pandas_dtype,
 )
 from pandas.core.dtypes.dtypes import (
+    CategoricalDtype,
     DatetimeTZDtype,
     ExtensionDtype,
     PeriodDtype,
-    CategoricalDtype
 )
 from pandas.core.dtypes.generic import (
     ABCDataFrame,
@@ -6455,25 +6455,27 @@ def astype(
 
         else:
             # else, only a single dtype is given
-            
-             # GH 61074: Make dtype="category" imply "ordered" = False + add deprecation warning
+
+            # GH 61074: Make dtype="category" imply "ordered" = False + add deprecation warning
             if dtype == "category":
                 if isinstance(self.dtype, CategoricalDtype):
                     if self.dtype.ordered:
                         warnings.warn(
                             "The 'category' dtype is being set to ordered=False by default.",
                             DeprecationWarning,
-                            stacklevel=3
-                            )
-                        
-                    if isinstance(dtype, CategoricalDtype):                
-                        dtype = CategoricalDtype(categories=dtype.categories ,ordered=False)
+                            stacklevel=3,
+                        )
+
+                    if isinstance(dtype, CategoricalDtype):
+                        dtype = CategoricalDtype(
+                            categories=dtype.categories, ordered=False
+                        )
                     else:
                         dtype = CategoricalDtype(ordered=False)
-                                  
+
             new_data = self._mgr.astype(dtype=dtype, errors=errors)
             res = self._constructor_from_mgr(new_data, axes=new_data.axes)
-            
+
             return res.__finalize__(self, method="astype")
 
         # GH 33113: handle empty frame or series

pandas/tests/series/methods/test_astype.py

@@ -610,19 +610,18 @@ def test_astype_categoricaldtype(self):
     def test_astype_categorical_to_categorical(
         self, name, dtype_ordered, series_ordered
     ):
-      
         def check_deprecation_warning(series):
-            ''' Helper function to check DeprecationWarning for ordered = True conversions'''
+            """Helper function to check DeprecationWarning for ordered = True conversions"""
             msg = "The 'category' dtype is being set to ordered=False by default."
-            with tm.assert_produces_warning(DeprecationWarning, match = msg):
+            with tm.assert_produces_warning(DeprecationWarning, match=msg):
                 result = series.astype("category")
             assert result.dtype.ordered is False
-            
+
         # GH#10696, GH#18593
         s_data = list("abcaacbab")
         s_dtype = CategoricalDtype(list("bac"), ordered=series_ordered)
         ser = Series(s_data, dtype=s_dtype, name=name)
-        
+
         # GH#61074
         if series_ordered is True:
             check_deprecation_warning(ser)
@@ -632,7 +631,7 @@ def check_deprecation_warning(series):
         # GH#61074
         # unspecified categories
         dtype = CategoricalDtype(ordered=False)
-        result = ser.astype(dtype)        
+        result = ser.astype(dtype)
         exp_dtype = CategoricalDtype(s_dtype.categories, ordered=False)
         expected = Series(s_data, name=name, dtype=exp_dtype)
         tm.assert_series_equal(result, expected)
@@ -641,7 +640,7 @@ def check_deprecation_warning(series):
         # different categories
         dtype = CategoricalDtype(list("adc"), False)
         result = ser.astype(dtype)
-        expected = Series(s_data, name=name, dtype=dtype)    
+        expected = Series(s_data, name=name, dtype=dtype)
         tm.assert_series_equal(result, expected)
 
         if dtype_ordered is False: