CLN: remove F821 flake8 error in test case (#22009)

pandas-dev · Jul 22, 2018 · 0828c25 · 0828c25
1 parent 37c7458
commit 0828c25
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pandas/tests/frame/test_dtypes.py b/pandas/tests/frame/test_dtypes.py
@@ -9,7 +9,7 @@
 import numpy as np
 from pandas import (DataFrame, Series, date_range, Timedelta, Timestamp,
                     Categorical, compat, concat, option_context)
-from pandas.compat import u, PY2
+from pandas.compat import u
 from pandas import _np_version_under1p14
 
 from pandas.core.dtypes.dtypes import DatetimeTZDtype, CategoricalDtype
@@ -356,9 +356,10 @@ def test_select_dtypes_datetime_with_tz(self):
         expected = df3.reindex(columns=[])
         assert_frame_equal(result, expected)
 
-    @pytest.mark.parametrize(
-        "dtype", [str, "str", np.string_, "S1",
-                  "unicode", np.unicode_, "U1"] + ([unicode] if PY2 else []))
+    @pytest.mark.parametrize("dtype", [
+        str, "str", np.string_, "S1", "unicode", np.unicode_, "U1",
+        compat.text_type
+    ])
     @pytest.mark.parametrize("arg", ["include", "exclude"])
     def test_select_dtypes_str_raises(self, dtype, arg):
         df = DataFrame({"a": list("abc"),