lf

Author: xaris96

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_old_base.py

@@ -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

@@ -13,7 +13,6 @@
 
 from pandas.core.dtypes.cast import find_common_type
 
-import pandas as pd
 from pandas import (
     CategoricalDtype,
     CategoricalIndex,
@@ -77,9 +76,9 @@ def test_union_different_types(index_flat, index_flat2, request):
     idx1 = index_flat
     idx2 = index_flat2
 
-    # mixed int string
-    if idx1.equals(pd.Index([0, "a", 1, "b", 2, "c"])) or idx2.equals(
-        pd.Index([0, "a", 1, "b", 2, "c"])
+    # Mixed int string
+    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)
@@ -244,7 +243,7 @@ def test_union_base(self, index):
         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:]
@@ -313,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]