Py2 compat

harisbal · Feb 8, 2018 · b20e12c · b20e12c
1 parent fbf0a06
commit b20e12c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pandas/tests/extension_arrays/test_common.py b/pandas/tests/extension_arrays/test_common.py
@@ -27,8 +27,10 @@ def test_astype():
 def test_astype_raises():
     arr = DummyArray(np.array([1, 2, 3]))
 
+    # type  int for py2
+    # class int for py3
     xpr = ("DummyArray can only be coerced to 'object' dtype, not "
-           "'<class 'int'>'")
+           "'<.* 'int'>'")
 
     with tm.assert_raises_regex(ValueError, xpr):
         arr.astype(int)