precommit check

Author: xaris96

pandas/tests/base/test_value_counts.py

@@ -66,7 +66,6 @@ def test_value_counts_null(null_obj, index_or_series_obj):
     if obj.dtype == "object":
         obj = obj.astype(str)
 
-
     values = obj._values
     values[0:2] = null_obj
 

pandas/tests/indexes/multi/test_setops.py

@@ -627,7 +627,7 @@ def test_union_with_duplicates_keep_ea_dtype(dupe_val, any_numeric_ea_dtype):
 @pytest.mark.filterwarnings(r"ignore:PeriodDtype\[B\] is deprecated:FutureWarning")
 def test_union_duplicates(index, request):
     # special case for mixed types
-    if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+    if index.equals(Index([0, "a", 1, "b", 2, "c"])):
         index = index.map(str)
 
     # GH#38977

pandas/tests/indexes/test_common.py

@@ -442,7 +442,6 @@ def test_hasnans_isnans(self, index_flat):
 @pytest.mark.xfail(
     reason="Sorting fails due to heterogeneous types in index (int vs str)"
 )
-
 def test_sort_values_invalid_na_position(index_with_missing, na_position):
     non_na_values = [x for x in index_with_missing if pd.notna(x)]
     if len({type(x) for x in non_na_values}) > 1:

pandas/tests/indexes/test_mixed_int_string.py

@@ -1,6 +1,8 @@
 import pytest
+
 import pandas as pd
 
+
 def test_mixed_int_string_index():
     idx = pd.Index([0, "a", 1, "b", 2, "c"])
 

pandas/tests/indexes/test_numpy_compat.py

@@ -156,13 +156,14 @@ def test_numpy_ufuncs_reductions(index, func, request):
     if len(index) == 0:
         pytest.skip("Test doesn't make sense for empty index.")
 
-    if any(isinstance(x, str) for x in index) and \
-            any(isinstance(x, int) for x in index):
+    if any(isinstance(x, str) for x in index) and any(
+        isinstance(x, int) for x in index
+    ):
         request.applymarker(
             pytest.mark.xfail(
                 reason=(
-                    "Cannot compare mixed types (int and str) in ufunc reductions"
-                    " and should raise a TypeError"
+                    "Cannot compare mixed types (int and str) in ufunc reductions "
+                    "and should raise a TypeError"
                 )
             )
         )

pandas/tests/indexes/test_old_base.py

@@ -359,7 +359,7 @@ def test_argsort(self, index):
             pytest.skip(f"{type(self).__name__} separately tested")
 
         # New test for mixed-int-string
-        if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+        if index.equals(Index([0, "a", 1, "b", 2, "c"])):
             result = index.astype(str).argsort()
             expected = np.array(index.astype(str)).argsort()
             tm.assert_numpy_array_equal(result, expected, check_dtype=False)
@@ -371,7 +371,7 @@ def test_argsort(self, index):
 
     def test_numpy_argsort(self, index):
         # new test for mixed-int-string
-        if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+        if index.equals(Index([0, "a", 1, "b", 2, "c"])):
             result = np.argsort(index.astype(str))
             expected = index.astype(str).argsort()
             tm.assert_numpy_array_equal(result, expected)

pandas/tests/indexes/test_setops.py

@@ -5,7 +5,7 @@
 
 from datetime import datetime
 import operator
-import pandas as pd
+
 import numpy as np
 import pytest
 
@@ -64,20 +64,22 @@ def index_flat2(index_flat):
 
 def test_union_same_types(index):
     # mixed int string
-    if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+    if index.equals(Index([0, "a", 1, "b", 2, "c"])):
         index = index.astype(str)
 
     idx1 = index.sort_values()
     idx2 = index.sort_values()
     assert idx1.union(idx2, sort=False).dtype == idx1.dtype
 
+
 def test_union_different_types(index_flat, index_flat2, request):
     idx1 = index_flat
     idx2 = index_flat2
 
     # Special handling for mixed int-string types
-    if idx1.equals(pd.Index([0, "a", 1, "b", 2, "c"])) or \
-            idx2.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+    if idx1.equals(Index([0, "a", 1, "b", 2, "c"])) or idx2.equals(
+        Index([0, "a", 1, "b", 2, "c"])
+    ):
         idx1 = idx1.astype(str)
         idx2 = idx2.astype(str)
 
@@ -130,6 +132,7 @@ def test_union_different_types(index_flat, index_flat2, request):
         assert res1.dtype == common_dtype
         assert res2.dtype == common_dtype
 
+
 @pytest.mark.parametrize(
     "idx1,idx2",
     [
@@ -234,14 +237,13 @@ def test_intersection_base(self, index):
 
     @pytest.mark.filterwarnings(r"ignore:PeriodDtype\[B\] is deprecated:FutureWarning")
     def test_union_base(self, index):
-
         if index.inferred_type in ["mixed", "mixed-integer"]:
             pytest.skip("Mixed-type Index not orderable; union fails")
 
         index = index.unique()
 
         # Mixed int string
-        if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+        if index.equals(Index([0, "a", 1, "b", 2, "c"])):
             index = index.astype(str)
 
         first = index[3:]
@@ -296,7 +298,6 @@ def test_difference_base(self, sort, index):
 
     @pytest.mark.filterwarnings(r"ignore:PeriodDtype\[B\] is deprecated:FutureWarning")
     def test_symmetric_difference(self, index, using_infer_string, request):
-
         if (
             using_infer_string
             and index.dtype == "object"
@@ -311,7 +312,7 @@ def test_symmetric_difference(self, index, using_infer_string, request):
             # index fixture has e.g. an index of bools that does not satisfy this,
             #  another with [0, 0, 1, 1, 2, 2]
             pytest.skip("Index values no not satisfy test condition.")
-        if index.equals(pd.Index([0, "a", 1, "b", 2, "c"])):
+        if index.equals(Index([0, "a", 1, "b", 2, "c"])):
             index = index.astype(str)
         first = index[1:]
         second = index[:-1]
@@ -920,8 +921,9 @@ def test_symmetric_difference_mi(self, sort):
         index2 = MultiIndex.from_tuples([("foo", 1), ("bar", 3)])
 
         def has_mixed_types(level):
-            return any(isinstance(x, str) for x in level) and \
-                any(isinstance(x, int) for x in level)
+            return any(isinstance(x, str) for x in level) and any(
+                isinstance(x, int) for x in level
+            )
 
         for idx in [index1, index2]:
             for lvl in range(idx.nlevels):

pandas/tests/test_algos.py

@@ -63,12 +63,9 @@ def test_factorize_complex(self):
         expected_uniques = np.array([(1 + 0j), (2 + 0j), (2 + 1j)], dtype=complex)
         tm.assert_numpy_array_equal(uniques, expected_uniques)
 
-    @pytest.mark.parametrize("index_or_series_obj",
-                             [
-                             [1, 2, 3],
-                             ["a", "b", "c"],
-                             [0, "a", 1, "b", 2, "c"]
-                             ])
+    @pytest.mark.parametrize(
+        "index_or_series_obj", [[1, 2, 3], ["a", "b", "c"], [0, "a", 1, "b", 2, "c"]]
+    )
     @pytest.mark.parametrize("sort", [True, False])
     def test_factorize(self, index_or_series_obj, sort):
         obj = Index(index_or_series_obj)
@@ -94,11 +91,9 @@ def test_factorize(self, index_or_series_obj, sort):
         if expected_uniques.dtype == bool and obj.dtype == object:
             expected_uniques = expected_uniques.astype(object)
 
-
         if sort:
             expected_uniques = expected_uniques.sort_values()
 
-
         # construct an integer ndarray so that
         # `expected_uniques.take(expected_codes)` is equal to `obj`
         expected_uniques_list = list(expected_uniques)