import compat.PY2 and compat.string_types directly

Author: janrito

pandas/util/testing.py

@@ -38,7 +38,7 @@
 import pandas.compat as compat
 from pandas.compat import (
     filter, map, zip, range, unichr, lrange, lmap, lzip, u, callable, Counter,
-    raise_with_traceback, httplib, StringIO, PY3)
+    raise_with_traceback, httplib, StringIO, string_types, PY3, PY2)
 
 from pandas import (bdate_range, CategoricalIndex, Categorical, IntervalIndex,
                     DatetimeIndex, TimedeltaIndex, PeriodIndex, RangeIndex,
@@ -993,15 +993,15 @@ def raise_assert_detail(obj, message, left, right, diff=None):
     elif is_categorical_dtype(left):
         left = repr(left)
 
-    if compat.PY2 and isinstance(left, compat.string_types):
+    if PY2 and isinstance(left, string_types):
         left = left.encode('utf-8')
 
     if isinstance(right, np.ndarray):
         right = pprint_thing(right)
     elif is_categorical_dtype(right):
         right = repr(right)
 
-    if compat.PY2 and isinstance(right, compat.string_types):
+    if PY2 and isinstance(right, string_types):
         right = right.encode('utf-8')
 
     msg = """{obj} are different