pre-commit

Author: topper-123

pandas/tests/arrays/categorical/test_map.py

@@ -32,6 +32,7 @@ def test_map_str(data, categories, ordered, na_action):
     )
     tm.assert_categorical_equal(result, expected)
 
+
 def test_map(na_action):
     cat = Categorical(list("ABABC"), categories=list("CBA"), ordered=True)
     result = cat.map(lambda x: x.lower(), na_action=na_action)
@@ -65,6 +66,7 @@ def f(x):
     result = cat.map({"A": 10, "B": 20, "C": 30}, na_action=na_action)
     tm.assert_categorical_equal(result, exp)
 
+
 @pytest.mark.parametrize(
     ("data", "f", "expected"),
     (
@@ -92,6 +94,7 @@ def test_map_with_nan_none(data, f, expected):  # GH 24241
     else:
         tm.assert_index_equal(result, expected)
 
+
 @pytest.mark.parametrize(
     ("data", "f", "expected"),
     (
@@ -119,6 +122,7 @@ def test_map_with_nan_ignore(data, f, expected):  # GH 24241
     else:
         tm.assert_index_equal(result, expected)
 
+
 def test_map_with_dict_or_series(na_action):
     orig_values = ["a", "B", 1, "a"]
     new_values = ["one", 2, 3.0, "one"]

pandas/tests/indexes/categorical/test_map.py

@@ -27,6 +27,7 @@ def test_map_str(data, categories, ordered):
     )
     tm.assert_index_equal(result, expected)
 
+
 def test_map():
     ci = CategoricalIndex(list("ABABC"), categories=list("CBA"), ordered=True)
     result = ci.map(lambda x: x.lower())
@@ -54,9 +55,7 @@ def f(x):
         return {"A": 10, "B": 20, "C": 30}.get(x)
 
     result = ci.map(f)
-    exp = CategoricalIndex(
-        [10, 20, 10, 20, 30], categories=[20, 10, 30], ordered=False
-    )
+    exp = CategoricalIndex([10, 20, 10, 20, 30], categories=[20, 10, 30], ordered=False)
     tm.assert_index_equal(result, exp)
 
     result = ci.map(Series([10, 20, 30], index=["A", "B", "C"]))
@@ -102,6 +101,7 @@ def test_map_with_nan_ignore(data, f, expected):  # GH 24241
     result = values.map(f, na_action="ignore")
     tm.assert_index_equal(result, expected)
 
+
 @pytest.mark.parametrize(
     ("data", "f", "expected"),
     (